PC:Huaweimatebook14
驱动:GeForce MX350
python:3.9
太长不看版:windows下cuda10.2安装pytorch gpu版本仅能使用1.10.1及之前版本,代码如下:
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=10.2 -c pytorch
最近由于实习需求要在笔记本上安装gpu版本的pytorch(公司给配的台式机还没到只能先在笔记本上实验一下,不得不在我的陈旧笔记本上去跑)。之前都是在服务器上跑代码所以自己机器上的很久没有用了,上次用还是一年前的大作业,之前的环境配的很乱,因此打算新建环境安装pytorch。
根据网上的指导在Nvidia控制面板中找到支持的cuda版本,我的是10.2,应该是只能安装低于这个版本的cuda。安装好后,去官网上查找之前的安装命令,如下:
到创建好的虚拟环境中输入命令进行安装(若安装失败则需要进行换源)。本以为一切大功告成,但是当按照传统三部曲去检查是否安装成功时(先在命令行对应环境中输入python再输入以下命令),最后一步却出现了False的错误。
import torch
print(torch.__version__)
print(torch.cuda.is_available())
开始调转方向去检查cuda和cudnn,按照链接方法http://t.csdn.cn/vrJGK检查后,发现没有问题。于是开始在网上瞎逛,偶然发现有文章说cuda10,2不再支持Windows,pytorch官网说明为(CUDA-10.2 PyTorch builds are no longer available for Windows, please use CUDA-11.3)当然这个说明只在10.2版本还在主页的时候能够看到(也是够坑,不支持为啥还要放链接(抓狂)。可是电脑驱动也装不了更高版本的cuda了呀,难道就不能用了吗?
不甘心地去尝试更早版本的Pytoch,安装了1.10.1的版本如下,发现是可用的!原来只是后边的版本不支持windows版本的cuda10.2了,但是官网里面却还是给了cuda10.2对应的linux和windows下载命令,有点迷惑住。
# CUDA 10.2
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=10.2 -c pytorch
# CUDA 11.3
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge
# CPU Only
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cpuonly -c pytorch
抱着严谨认真的态度,又去测试了一下其他版本,发现1.11.0及之后的都不行,所以uu们如果是相同情况只有cuda10.2并且要装pytorchgpu版本的话就使用上方代码就好。