RGB-D代码运行错误
RuntimeError: No CUDA GPUs are available
测试:
print(torch.cuda.device_count())
print(torch.cuda.is_available())
输出:
0
False
# 若显示 1 TRUE,则在代码前加入torch.cuda.empty_cache()语句可解决,(来源网络,未验证)
修改代码:
os.environ["CUDA_VISIBLE_DEVICES"] = '0'