首先:我们要安装cuda toolkit
nvidia官网下载(国内可能进不去)
nvcc -V
cmd输入以上代码检测是否安装成功
然后是配置环境:
首先是
import torch
print(torch.cuda.is_available()) # 是否可以用gpu False不能,True可以
print(torch.cuda.device_count()) # gpu数量, 0就是没有,1就是检测到了
print(torch.__version__) #结果X.X.X+cpu不行
进入官网Start Locally | PyTorch下载pytorch
#torch.cuda.is_available()返回false——解决办法_cuda available false-CSDN博客这篇文章有更详细的解决办法 建议查看