毕设要使用pytorch,因为之前下载过tensoflow所以知道要找对应版本,就先看了一下自己需要下载的版本:
查询链接:https://pytorch.org/get-started/locally/#no-cuda-1
从这个图可以看到我要下载的是cuda11.3。
可以直接在anaconda prompt激活要安装pytorch的虚拟环境,通过以下语句安装cuda及cudnn。
# 安装CUDA
conda install cudatoolkit=11.3 # 指定版本
# 安装cudnn,如果不指定版本,在安装CUDA之后,会自动匹配对应版本的cudnn安装
conda install cudnn
出现问题:
安装完cuda和cudnn开始安装pytorch,刚开始我直接复制了上面图片中的语句:conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
但是安装完成使用
torch.cuda.is_available()
验证时会出现False。
差了百度之后发现使用这个语句默认安装cpu版本的pytorch,要下载gpu版本pytorch需要使用pip
下载。
也就是:
但是直接使用:
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
会因为torch文件下载太慢导致下载失败。
解决方案:
可以直接进入上述语句后缀的网站下载相应torch、torchvision、torchaudio的whl文件:
https://download.pytorch.org/whl/cu113
查了很多帖子发现之前的官网语句直接标明了需要下载的版本,但是现在官网给的语句没有版本号。
但是我发现,使用以上语句下载时后面会直接跟上版本号:
在网站中找到对应版本号的whl文件并且下载
下载完成后在cmd
(不是anaconda prompt,在这里还是回超时)进入whl文件的下载目录,并运行:
pip install torch-1.11.0+cu113-cp39-cp39-win_amd64.whl
注意:这条语句会把pytorch安装到计算机默认的python环境中,也就是环境变量里在path中的文件位置。所以想要把pytorch安装到你要用的python环境(不是默认python环境)中有两个办法:
一、在默认python环境中安装好pytorch后,在默认python环境/Lib/site-packages中找到pytorch的两个文件夹,把他们剪切到你要用的python环境/Lib/site-packages中。
二、把环境变量path中原有的python路径删掉,添加你要使用的python路径。
安装完成后回到anaconda prompt,激活虚拟环境并再次使用:
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
或
pip3 install torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
之前下载失败是由于torch文件太大了,而torchvision和torchaudio都比较小,所以可以正常下载。
完成之后使用
python
import torch
torch.cuda.is_available()
如果出现True,那么恭喜你安装成功辣!!!