安装gpu版本的pytorch

参考博主从0到1教你如何安装Pytorch【CPU版&GPU版】_pytorch cpu版本-CSDN博客

1. 设备管理器 查看有无英伟达显卡

2.有的话在cmd命令行输入nvidia-smi看cuda版本，我的是11.6

3.安装cuda CUDA Toolkit Archive | NVIDIA Developer，注意安装cuda不能超过这个版本，我选的11.4

4. 安装pytorch GPU pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113

5.命令行 进入python环境，输入

import torch

print(torch.__version__)

print(torch.cuda.is_available())

检查GPU是否安装成功