1.下载镜像(镜像网址:https://download.pytorch.org/whl/torch_stable.html)
假设你要下载torch1.4.0版本,cp36代表你的环境是python3.6,cu100代表的是你的cuda是10.0版本
2.查看torch1.4.0对应的torchvision版本
Previous PyTorch Versions | PyTorch
3.已知torch1.4.0对应torchvision0.5.0
4.安装打开项目文件的终端
分别输入(要放在你的项目文件中,终端前面目录也要对)
pip install torch-1.4.0+cu100-cp36-cp36m-linux_x86_64.whl pip install torchvision-0.5.0+cu100-cp36-cp36m-linux_x86_64.whl
5.完成
(最后安装完了才知道,虚拟机不能用cuda。。)
ubuntu linux安装pytorch和torchvision
于 2023-04-13 11:39:46 首次发布