精彩文章免费看

No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda'

场合:训练模型的时候报错,无法继续进行

训练模型的不同epoch间报错,不影响程序运行

print(torch.cuda.is_available()),显示false(包括nvidia-smi也不好使,报错如下)

Ubuntu中Failed to initialize NVML: Driver/library version mismatch问题的解决_Explore IT & Research-CSDN博客


解决:目前来看重启就解决了

最后编辑于:2021-01-26 23:37