如何安装cuda版本的pytorch

为什么安装Cuda

对于做深度学习研究的小伙伴本，当我们处理大量的数据时，尤其是图像数据时，过量的数据会导致我们的CPU运行压力过大，占用大量的运行内存，而且用CPU进行模型训练，训练的时间会很长，这不利于我们PC用户的科学研究，相反，Cuda是NVIDIA开发出来的一个用于模型训练的一个很好的工具，可以快速有效的帮助我们进行模型的训练。

安装Cuda的流程

检查我们PC支持的Cuda版本

win + R 打开我们的运行管理器输入cmd，打开命令行，输入下面这行代码

nvidia -sim

我这里支持CUDA的版本是12.6

安装Cuda

这里附上官网链接

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

大家通过复制进行搜索

打开网址是这样的，你根据你支持的Cuda版本下载相应的Cuda

安装的过程一路点击确定即可，对于我们普通用户选择精简版就可以

Cuda 安装验证

同样的，在命令行中输入

nvcc -V

产生和下面这个图片差不多的信息就表明你安装成功了

Pytorch 的Cuda版本的安装

直接在我们的python终端执行

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/

在whl后面加上你的cuda版本比如cuda11.8

这样你就可以下载成功了