windows10环境下安装anaconda+pytorch+CUDA10.2+cuDDN

anaconda安装就不说了,官网下载安装即可
下载地址:https://www.anaconda.com/download/
.
.
.
步骤一:安装CUDA

步骤二:安装cuDDN

步骤三:测试
.
.
.

一:安装CUDA

NVIDIA控制面板→系统信息->组件,查看NVCUDA.DLL的版本
在这里插入图片描述
在这里插入图片描述
然后进入CUDA官网下载对应版本的CUDA:
https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
(注意这里查询本机CUDA版本后可以安装低版本的CUDA,我的电脑查询后是CUDA11.0,然后我安装的10.2版本的CUDA)
在这里插入图片描述
可以选择自定义安装,一路默认即可。
在这里插入图片描述
在这里插入图片描述
安装完毕(我自己的过程没保存,图片来源网络)
启动cmd命令行,然后输入:

cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin

(我安装的是10.2版本)
然后输入:

nvcc -V 

注意是大写的“V”,如下图所示安装成功。
在这里插入图片描述
.
.
.

二:安装cuDDN

下载cuDDN的网址:
https://developer.nvidia.com/rdp/cudnn-download
需要注册一个NVIDIA账号,然后如下图进行选择:
在这里插入图片描述
下载完成后,解压:
在这里插入图片描述
然后将这三个文件夹下在这里插入图片描述
下的文件分别复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2目录中对应的bin、include、lib目录中。
最后打开“此电脑”→“高级系统设置”→“环境变量”→“系统变量”→“path”查看是否存在该bin目录,没有的话新建添加即可。
.
.
.

三:测试:

在这里插入图片描述
和我的输出一样,就是大功告成了!
.
.
.

If

这里你的输出出错了,检查一下你的pytorch是否是不带CUDA版本,可以先删除pytorch重新安装:
删除命令:

pip uninstall torch

然后进入pytorch官网选择合适的版本:
https://pytorch.org/get-started/locally/

这里建议选择pip方式安装,conda方式我安装的时候报乱七八糟的错,我当时直接在pycharm中的terminal中运行的command,运行命令是:

pip install torch===1.5.0 torchvision===0.6.0 -f https://download.pytorch.org/whl/torch_stable.html

大约一个半小时吧,安装完成:
在这里插入图片描述
ok,结束,希望可以帮到大家~

猜你喜欢

转载自blog.csdn.net/Su_Del/article/details/106530652