RuntimeError: No CUDA GPUs are available【RSCNN复现】

项目场景：

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.csdn.net/qq_43350820/article/details/110261939