项目场景:
RSCNN复现
RuntimeError: No CUDA GPUs are available
问题描述
RuntimeError: No CUDA GPUs are available

原因分析:
查看是否有GPU,以及GPU的设备数量:
print(torch.cuda.device_count())
print(torch.cuda.is_available())

解决方案:
修改如下
CUDA_VISIBLE_DEVICES=0

参考:https://blog.youkuaiyun.com/qq_43350820/article/details/110261939