在windows中安装pytorch的GPU版本及torchvision

一. 查看自己电脑的CUDA版本

在powershell中输入:nvidia-smi

nvidia-smi

 当前CUDA版本为12.6，表示支持最高版本的CUDA是12.6，可以向下兼容低版本的CUDA

二. 下载安装CUDA

NVIDIA官网： https://developer.nvidia.com/cuda-toolkit-archive

用上面的网址下载适配的版本。我用的是windows11，用11.6这个版本。

 

 下载好之后默认安装即可

三. 查看环境变量

安装完成后查看系统变量：

在“系统属性”--“环境变量”中查看

 在powershell输入nvcc --version验证安装

nvcc --version

 

四. 安装GPU版本的Pytorch和torchvision

pytorch官网的老版本安装链接：https://pytorch.org/get-started/previous-versions/

 我用pycharm做IDE，是用pip安装，复制对应这行命令安装。其它的开发环境对应复制官网的命令。

 也可以用离线下载的方式下载安装，Pytorch下载链接：https://download.pytorch.org/whl/torch_stable.html.

在powershell中输入pip list验证

pip list

 验证开发环境

 若 torch.cuda.is_available() 结果为True，则表示GPU版本安装完成