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



