安装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
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