解决torch.cuda.is_available()一直为false的问题

一、检测脚本

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

输出

二、解决排查

1、安装CUDA

根据驱动选择要安装的版本

安装网站：CUDA Toolkit Archive | NVIDIA Developer

安装成功后

2、安装cuDNN

注意：需要注册才能进行安装

具体网站：cuDNN 9.1.1 Downloads | NVIDIA Developer

下载后，需要解压

需要将bin、include、和lib/x86目录中的文件拷贝到亲们cuda安装的对应目录中

并且新增如下环境变量：

配置成功后，使用nvidia-smi命令检查如下

3、修改pytorch版本，从cpu版本改为gpu版本

1）卸载cpu版本

pip3 uninstall torch torchvision torchaudio

2）安装gpu版本

网站：Start Locally | PyTorch

使用网站上面生成的命令进行安装

三、验证成果

执行文章最开始的脚本，结果如下