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博客


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

你可能感兴趣的:(No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda')