torch.cuda.is_available()返回flase 问题排查

查看torch版本是否为GPU版本

import torch
print(torch.__version__)    //我的是CPU版本

如果不是GPU版本下载GPU版本

• 查看cuda版本

nvcc --version


nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Wed_Sep_21_10:41:10_Pacific_Daylight_Time_2022
Cuda compilation tools, release 11.8, V11.8.89
Build cuda_11.8.r11.8/compiler.31833905_0    //我的是11.8版本

• 下载与cuda版本一致的pytorch

//下载pytorch 11.8版本，去pytorch官网生成11.8对应的pytorch版本

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118