python - CUDA 初始化 : Unexpected error from cudaGetDeviceCount()

Question

我在我的 Linux 服务器上运行一个深度学习程序，突然出现这个错误。UserWarning: CUDA initialization: Unexpected error from cudaGetDeviceCount(). Did you run some cuda functions before calling NumCudaDevices() that might have already set an error? Error 804: forward compatibility was attempted on non supported HW (Triggered internally at /opt/conda/conda-bld/pytorch_1603729096996/work/c10/cuda/CUDAFunctions.cpp:100.)早些时候，当我刚刚创建这个 conda 环境时，torch.cuda.is_available()返回 true我可以使用 CUDA 和 GPU。但突然间我无法使用 CUDA 和 torch.cuda.is_available()返回 false .我该怎么办？
附:我使用 GeForce RTX 3080 和 cuda 11.0 + pytorch 1.7.0。它以前有效，但现在没有了。

Answer 1

我刚刚尝试重新启动。问题解决了。原来是 NVIDIA NVML 驱动程序/库版本不匹配导致的。