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博客
解决:目前来看重启就解决了