安装CUDA

输入命令,查看我的Driver Version:551.76和CUDA Version:12.4。

// An highlighted block
nvidia-smi

在这里插入图片描述
去官网查看是否满足要求:链接: NVIDIA CUDA

在这里插入图片描述
只要安装的CUDA 不超过上限即可,我选择的是11.6版本的CUDA。
地址:链接: NVIDIA CUDA 11.6
在这里插入图片描述
下载后默认安装
在这里插入图片描述
先在自己电脑上安装软件的盘中先建立两个文件夹,这个路径一定要和之前安装exe文件的路径不一样。
在这里插入图片描述
将第一个和第三个目录更换为上面的CUDAToolkit,第二个目录更换为CUDASamples。
在这里插入图片描述
系统环境变量添加NVCUDASAMPLES11_6_ROOT和NVCUDASAMPLES11_6_ROOT
在这里插入图片描述
双击Path
在这里插入图片描述
添加两个路径。
在这里插入图片描述
在cmd中输入

nvcc -V

出现如下即安装成功。
在这里插入图片描述

安装pytorch

地址: CUDA11.6版本的pytorch
在这里插入图片描述

pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116

更多推荐