[ERROR] RuntimeError: No CUDA GPUs are available

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'

你可能感兴趣的:(笔记,pytorch,深度学习)