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编号
- 在终端中运行python程序时设置
CUDA_VISIBLE_DEVICES=0 python main.py
- 在python代码中设置
import os
os.environ['CUDA_VISIBLE_DEVICE']='0' # 使用编号为0的显卡
- 在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"
如果还是不行的话，可能需要重新装驱动了