环境

window 11
Python: 3.10.11

测试是否支持GPU

import torch

print(torch.__version__)
print(torch.cuda.is_available())

如果 torch 返回 X.X.X+cpu，则无法调用GPU，只会调用cpu，需要安装带有 cuda 的版本, 如下

2.0.1+cpu

这种只支持cpu

安装带有cuda版本之前，先查看 N 卡的驱动版本

nvidia-smi

安装带有cuda的torch包

查看cuda对应的版本
https://pytorch.org/get-started/previous-versions/

查找带有cuda的torch安装包
https://download.pytorch.org/whl/torch_stable.html

以下是 cuda 11.7 的安装命令行：

pip install torch==2.0.1+cu117 torchvision==0.15.2+cu117 torchaudio==2.0.2+cu117 -f https://download.pytorch.org/whl/torch_stable.html

安装后如果报错，找不到 torch 模块

可以再安装一次 torch

pip install torch

完整文章教程

https://blog.csdn.net/qq_46126258/article/details/112708781