Cuda+PythorchGPU版本安装

CUNDA

下载

查看自己电脑显卡的Cuda版本，命令行执行命令：

nvidia-smi

确定自己能使用的CUDA版本， 即CUDA Runtime Version，也就是上面图上的版本。要保证CUDA Driver 版本 >= CUDA Runtime 版本，像我的CUDA是12.3版本，我安装的时候选择了11.8的版本。
接下去英伟达官网下载11.8版本的Cuda。
CUDA Toolkit 11.8 Downloads

点击下载。

安装

安装cuda时，第一次会让设置临时解压目录，第二次会让设置安装目录；
临时解压路径，建议默认即可，也可以自定义。安装结束后，临时解压文件夹会自动删除；
安装目录，建议默认即可；
:::info
临时解压目录不要和cuda的安装路径设置一样，否则安装结束，会找不到安装目录！此处我将临时安装目录设置到非系统盘路径，但是注意要选择空文件夹，否则会报错。
:::





自定义安装，精简版本是下载好所有组件，并且会覆盖原有驱动，所以在这里推荐自定义下载。

如果你是第一次安装，尽量全选
如果你是第n次安装，尽量只选择第一个，不然会出现错误。







打开控制面板即可发现安装了一大堆配置和驱动文件。

环境变量

win+R键运行cmd，输入nvcc --version 即可查看版本号；
set cuda，可以查看 CUDA 设置的环境变量。

Pythorch

Start Locally

在你的虚拟环境下执行命令

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

接着解释静静等待了

import torch
torch.cuda.is_available()

输出true安装成功