如何快速安装Pytorch1.10+cuda11.1

 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的“下载”文件夹中。

如何快速安装Pytorch1.10+cuda11.1_第1张图片

⑤ 在“下载”文件夹下,执行如下两个命令:

pip install torch-1.10.0+cu111-cp38-cp38-linux_x86_64.whl
pip install torchvision-0.11.0+cu111-cp38-cp38-linux_x86_64.whl

 ⑥ 安装很快就完成了。

 

 ⑦ 成功!!!!!!!!!!!!!!!!!

如何快速安装Pytorch1.10+cuda11.1_第2张图片

你可能感兴趣的:(深度学习环境配置,pytorch,深度学习,神经网络,机器学习,人工智能)