Win10+MX250+CUDA11.3+cuDNN+Pytorch安装

疫情在酒店不想远程控制台式,自己用华为电脑GPU: mx250

查了下英伟达的驱动和cuda的对应关系。为了防止别人的博客消失,我这里保存一下。

明确英伟达驱动和cuda不是一个。英伟达驱动(我的版本是441.6更新到最新的是500),控制面板查看的是驱动对应的cuda版本如下图,下载cuda工具包11.3。

cuda和英伟达驱动对应的版本(一般cuda都向下兼容,这里11.3的应该是也可以):https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

1、查看英伟达控制面板-系统信息-组件-Cuda版本10.2;

Win10+MX250+CUDA11.3+cuDNN+Pytorch安装_第1张图片Win10+MX250+CUDA11.3+cuDNN+Pytorch安装_第2张图片

pytorch win10不支持10.2cuda,支持11.3,因此到第二步更新英伟达的驱动才能安装高版本的cuda工具包。

2、更新英伟达驱动:GeForce Experience-->驱动程序--->更新驱动;

Win10+MX250+CUDA11.3+cuDNN+Pytorch安装_第3张图片

更新驱动后再查英伟达面板,cuda版本改变,向下兼容。

Win10+MX250+CUDA11.3+cuDNN+Pytorch安装_第4张图片

显然441小于452,不支持11.3,更新到最新的就可以了,只要大于就行!!虽然下面写的11.6 

Win10+MX250+CUDA11.3+cuDNN+Pytorch安装_第5张图片

如果一定要觉得11.6有点不妥,希望搞11.3,可以到官网去下载对应452以上的驱动。

英伟达Game驱动和Studio驱动有什么区别?其实Game驱动针对游戏玩家安装,为各大游戏提供绝佳的游戏体验。studio字面就是工作室,原来叫creator driver,适合CG行业方面的从业人员,推荐安装Studio版本,在稳定性和三维软件的针对性上有优化加成吧。

到官方下载英伟达驱动 https://www.nvidia.cn/geforce/drivers/

 3、安装pytorch,到官网pytorch.org,点这里https://pytorch.org/get-started/locally/

我的电脑本身就有anaconda,没有的自己先下载一个,conda命令好用推荐

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

4、点这里下载cuda11.3,点这里下载cudnn8.2

完成!

5、pytorch下载太慢

(1)更换镜像源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

在命令行中输入下述命令:

conda install pytorch torchvision cudatoolkit=11.3

5、验证cuda是否装好:cmd-> nvcc -V

Win10+MX250+CUDA11.3+cuDNN+Pytorch安装_第6张图片

验证pytorch

#判断是否安装了cuda
import torch
print(torch.cuda.is_available())  #返回True则说明已经安装了cuda
#判断是否安装了cuDNN
from torch.backends import  cudnn 
print(cudnn.is_available())  #返回True则说明已经安装了cuDNN

Win10+MX250+CUDA11.3+cuDNN+Pytorch安装_第7张图片

 

原文链接:https://blog.csdn.net/RufyHuang/article/details/106572460

你可能感兴趣的:(python)