RuntimeError: No CUDA GPUs are available

问题描述

使用CUDA进行模型训练时出现错误
RuntimeError: No CUDA GPUs are available

问题解决

使用CUDA之前检测是否有GPU，没检测就默认没有

print(torch.cuda.is_available())
©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容