Pytorch1.10 更新了有几天,正好最近项目需要,所以尝试更新了一下从Pytorch1.8到Pytorch1.10。
本文假设已经安装好cuda11.1的环境。
①
http://download.pytorch.org/whl/torch_stable.html
② 搜索torch1.10 (注意操作系统,GPU/CPU,cuda版本和python版本,作者是linux+cuda11.1+python3.8所以就选了如下的文件,点击下载)
③ 搜索与pytorch1.10相匹配的torchvision 0.11.0(如下图所示)
④ 将二者下载到相同的文件夹中,作者直接下载到Ubuntu18.04的“下载”文件夹中。
⑤ 在“下载”文件夹下,执行如下两个命令:
pip install torch-1.10.0+cu1