CUDA安装教程以及torch torchvision torchaudio的安装

写在前面

CUDA分为Gpu版本以及Cpu版本，也就是你的电脑有无显卡，这里我们以安装有显卡的Gpu版本为例子。

官方教程：CUDA：Installation Guide Windows :: CUDA Toolkit Documentation

第一步、查看自己的Gpu支持的cuda版本

1、按住Win+R然后输入cmd进入命令提示符界面

2、在该界面输入指令nvidia-smi

第二步、下载cuda

1、进入cuda下载官网

CUDA Toolkit Archive | NVIDIA Developer

2、选择版本

选择好版本后按系统进行选择，这里按如图选择

3、安装

先选择合适的安装路径

这里选择自定义安装

首次安装可全部勾选

点击下一步等待安装完成即可

4、查验是否安装成功

再次打开命令提示符，可以在命令行中运行nvcc -V命令，如果显示CUDA版本信息，则说明安装成功。

第三步、安装torch torchvision torchaudio库

1、前往官网选择合适的版本

https://download.pytorch.org/whl/cu118

2、选择版本

要选择与自己cuda和python相兼容的版本

3、下载与安装

下载完成后记住下载文件的位置以及名字

打开命令提示符采用pip install + 文件位置+文件名的方式安装

同样的，torchvision、torchaudio与torch安装方法相同

在其中找到合适的版本

4、查验是否安装完成

如果安装成功在python解释其中可以看它们