最开始，创建环境，用这个代码查看显卡 

import torch
print(torch.cuda.is_available())
print(torch.version.cuda)
print(torch.cuda.get_device_name(0))

返回都是false

修改方法：

用这个跑在创建好的环境里 

conda install pytorch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 pytorch-cuda=12.1 -c pytorch -c nvidia

再跑第一个代码，就能找到