一.1.判断自己需要的版本。
首先看一下自己电脑能能用的最高配是哪个版本的,但是最新最高的不一定合适,据说向下兼容可以用最新的,我的是12.0版本,我下载了11.8版本的,正常使用。
参考教程:CUDA安装教程(超详细)_Billie使劲学的博客-CSDN博客
2.我修改了安装路径,安装在了D盘,注意这个路径还要添加环境变量时使用。
安装时候观察“新版本”和“当前版本”如果,当前版本高或者持平就不用勾选,另外Visual Studio Integration不用选,会出错。
按步骤来就行了,没啥坑。
二.
1.https://developer.nvidia.com/rdp/cudnn-download cuDNN的官网,查找匹配的版本就好了。
2.下载后是一个压缩包,将里边的文件复制到相应的文件夹之中,有相同的就覆盖。相应的文件夹就是刚才保存路径的文件夹。
没什么坑了
三.安装pytorch
官网地址Start Locally | PyTorch选择合适的版本,实际上我是根据pytorch来决定cuda的。
打开创建的虚拟环境,输入红框内的pip命令。
就开始下载了,等待完成,查看 命令(pytorch) D:Anaconda>conda list
这样就显示成功了。