CUDA下载,以及下载GPU版本的pytorch

一、下载anaconda

因为这步我之前就下好了,主要参考这个链接:史上最全最详细的Anaconda安装教程

二、下载CUDA

1.首先观察自己需要什么版本的CUDA,以及是否安装过CUDA
先cmd,输入命令

nvidia-smi

结果如下,所以我们可以下载12.1版本的CUDA

CUDA下载,以及下载GPU版本的pytorch_第1张图片然后输入命令

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

  1. 我遇到的问题
    在安装完CUDA之后,再次在cmd中输入 nvcc -V 仍然显示“nvcc‘ 不是内部或外部命令,也不是可运行的程序或批处理文件”,搜索后发现是环境变量没有配置好的原因,解决方法在这篇博客里面,配置好之后再次 nvcc -V就成功啦。

三、下载GPU版本的pytorch

1.安装方法也比较简单,参考这篇博客,我使用的是文中的第二种:conda安装。
2.不知道怎么新建conda环境的,可以看这篇博客

你可能感兴趣的:(pytorch,python,深度学习)