RuntimeError: No CUDA GPUs are available问题解决

RuntimeError: No CUDA GPUs are available问题解决

检查GPU是否可用

import torch
 
if torch.cuda.is_available():
    print("GPU可用")
else:
    print("GPU不可用")
 

显示当前可用的GPU数量

import torch
 
print("当前可用的GPU数量: ", torch.cuda.device_count())
 

Pytorch设置GPU编号

1. 在终端中运行python程序时设置

CUDA_VISIBLE_DEVICES=0 python main.py

2. 在python代码中设置

import os
os.environ['CUDA_VISIBLE_DEVICE']='0' # 使用编号为0的显卡

3. 在PyTorch代码中使用函数torch.cuda.set_device设置

import torch
 
# 设置使用编号为0的GPU
torch.cuda.set_device(0)
 

解决办法

os.environ['CUDA_VISIBLE_DEVICES'] 指定了笔记本不存在的显卡编号，你改成你自己的设备号就行啦。比如说我的是0，那么将os.environ['CUDA_VISIBLE_DEVICES'] = "0"

如果还是不行的话，可能需要重新装驱动了