之前运行正常突然报错:RuntimeError: No CUDA GPUs are available

// 报错信息
RuntimeError: No CUDA GPUs are available

解决方法

// 检测一下cuda是否可用
print(torch.cuda.is_available())
False
//重新在文件头加入下面说明
import torch
torch.cuda.current_device()
torch.cuda._initialized = True
Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐