1查看python对应的pytorch版本
pytorch的github官方文档:
https://github.com/pytorch/vision#installation
只要根据自己的python版本,选择torch就行,torchvision在之后用pytorch的官网的命令可以直接确定(第二步会讲)
近期pytorch版本:
往期pytorch版本:
2查看CUDA版本
输入命令nvcc -V
nvcc -V
如图,我的CUDA版本为11.1
注:命令nvidia-smi查看的是自己电脑显卡驱动下支持的CUDA最高版本,不是当前电脑上有的CUDA版本!如下图,我的电脑能支持的最高版本为11.6,那么我安装的CUDA版本不能超过这个数,而我安装的CUDA版本为11.1,没有超过。
nvidia-smi
3 查看CUDA对应的pytorch版本安装
还记得我们在第一步查看了python对应的pytorch版本吗,也就是说,要安装的pytorch必须同时满足python版本和CUDA版本,两者取交集哦
先看链接,还不急着安装
pytorch最新版本:https://pytorch.org/get-started/locally/
pytorch往期版本:https://pytorch.org/get-started/previous-versions/
4使用pip本地安装
注:直接把官网的命令粘贴到命令行可能安装不成功,我的情况是会报WARNING,并显示找不到相应版本,网上看有的网友们似乎anaconda虚拟环境能成功。因为此处不用conda,因此通过wheel进行安装。
如图,根据CUDA版本找到相应命令
采用wheel方式安装,需要每个库单独下载对应的wheel,分别安装,黄框内标注了此次要下载的三个库:torch torchvision trochaudio
接下来以torch为例进行安装
(1)下载wheel
下载地址:
https://download.pytorch.org/whl/cu111/
推荐迅雷下载,一般浏览器下载会比较慢
(2)pip install安装torch
pip install 【刚刚下载的torch的wheel的地址】
重复上述两小步,安装剩下的torchvision、torchaudio
5 检测pytorch是否安装成功
cmd输入pip install
pip install
再分别输入以下命令
python
import torch
torch.cuda.is_available()
如上图,出现True就算安装成功啦!