CUDA下载,以及下载GPU版本的pytorch
一、下载anaconda
因为这步我之前就下好了,主要参考这个链接:史上最全最详细的Anaconda安装教程
二、下载CUDA
1.首先观察自己需要什么版本的CUDA,以及是否安装过CUDA
先cmd,输入命令
nvidia-smi
结果如下,所以我们可以下载12.1版本的CUDA
然后输入命令
nvcc -V
如果输出的是
“nvcc”不是内部或外部命令,也不是可运行的程序或批处理文件
那就说明没安装CUDA(或者是下载过但是没配置好环境变量哈哈)
2.安装CUDA
这个链接非常详细了,基本跟着做就没问题:cuda安装 (windows版)
记录一下我安装的版本:
cuda_12.1.0_531.14_windows.exe
cudnn-windows-x86_64-8.9.0.131_cuda11-archive.zip
- 我遇到的问题
在安装完CUDA之后,再次在cmd中输入 nvcc -V 仍然显示“nvcc‘ 不是内部或外部命令,也不是可运行的程序或批处理文件”,搜索后发现是环境变量没有配置好的原因,解决方法在这篇博客里面,配置好之后再次 nvcc -V就成功啦。
三、下载GPU版本的pytorch
1.安装方法也比较简单,参考这篇博客,我使用的是文中的第二种:conda安装。
2.不知道怎么新建conda环境的,可以看这篇博客