如何安装GPU版本的pytorch

附1：配置清华源

如果没有清华源，可以配置一下(这样快一点）

conda config --show channels

conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2

（把这些链接都复制就好了）

第一步:查看CUDA版本

先查看自己的电脑显卡支持的CUDA版本，打开命令提示符输入

nvidia-smi查看

红框里的就算电脑适应的版本，或者你可以通过这个更新CUDA版

点击下载可以更新你的驱动版本

第二步：CUDA下载安装

CUDA下载安装

https://developer.nvidia.com/cuda-toolkit-archive

我的更新了，我的CUDA版本是12.4，所以我下载了12.4版本的，根据自己的版本选择点进去下载

下载到指定的文件夹，运行exe文件（这个文件夹是临时文件夹，安装重启后会被自动删除，所以后面安装不能在这个文件夹内，最好在文件夹里新建其它文件夹，以防万一不小心被一起删掉了，造成后面的运行报错）自己选择安装路径，由于我安装过了，我就不再安装，就去网上找个图放这，步骤都是一样的.

一直点下一步就好了，安装后就查看环境变量，按快捷键win+R后进入环境变量，查看环境变量是否添加完成

重启电脑让环境变量审生效，重启之后cmd打开控制台，输入nvcc -V（注意是大写的V）查看安装是否成功，如下图所示就说明安装成功了

如果出现安装之后无法找到nvcc命令，说不是内部命令，就需要打开path路径，手动添加环境变量，每个人的路径不一样，根据自己的去添加。我的如下图所示：

第三步：CUDNN下载

cudnn下载安装https://developer.nvidia.com/cudnn

进入cudnn官网，对应自己的CUDA版本去下载对应版本的cudnn

下载之后解压，复制这三个文件夹到指定文件夹（直接ctrl c+v就好）

之后添加环境变量，有点电脑需要，有的电脑不需要（我的不需要）需要的话就类似整个图一样去添加，我的不需要，所以我是借鉴别人的图，需要的话就把刚刚三个文件夹的路径添加到path中

第四步：创建虚拟环境，下载pytorch

进入然后进入Anaconda Prompt环境，再里面新建一个环境，由于是创建一个虚拟的环境所以和你自己电脑实际下载的Python版本没有关系的，不依赖与你电脑本地原来安装的Python版本。

conda create -n 你的环境名 python=版本号（这个版本号和你本地安装的python版本是没有关系的）

##比如我就写的3.10

conda create -n pytorch python=3.10

直接回车就好，出现这个就点击y

等待创建就好，安装成功之后就会显示

输入conda activate pytorch可以看见base，变成了pytorch，环境变更成功

去pytorch官网下载自己pytorch对应的CUDA版本，不能高于电脑查到的版本

复制你对应版本的这个指令去你刚刚创建的环境中运行就好，遇到就输入y就好  

安装成功之后，就输入pip list 出现了这个就说明安装成功

最后一步，完整的验证一下是否安装成功

出现了True说明安装成功，到此结束。