安装pytorch-gpu版本

安装pytorch1.7.1(GPU版本)

conda换清华源

1
2
3
4
conda config --remove-key channels
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64
conda config --set show_channel_urls yes

conda创建一个python=3.7的虚拟环境,并激活

1
2
conda create -n test python=3.7
activate test

进入该网址下载离线版本的pytorch(gpu) https://download.pytorch.org/whl/torch_stable.html

还有torchversion

(cu102指的是cuda的版本为10.2的,百度查询自己的cuda版本,然后下载对应版本即可,如果没有对应版本,下载比自己cuda版本稍低一点的即可)

在激活的test虚拟环境中完成下面的操作

输入命令:pip install (whl文件保存路径)\文件名称

我刚才下载的两个文件放到了桌面上,所以我的命令如下(改为自己的存储路径即可)

1
2
pip install C:\Users\lwd\Desktop\torch-1.7.1-cp37-cp37m-win_amd64.whl
pip install C:\Users\lwd\Desktop\torchvision-0.8.2-cp37-cp37m-win_amd64.whl

安装完毕后

输入以下命令检查是否安装正确,如图所示打印为True说明安装正确!


本博客所有文章除特别声明外,转载请注明出处!