ubuntu系统5090的显卡安装pytorch的GPU提示错误CUDA driver initialization failed, you might not have a CUDA gpu

在linux系统ubuntu系统里安装的GPU的pytorch测试GPU时提示如下错误信息。PyTorch CUDA状态: False。

纬领网络

275人浏览 · 2025-06-23 22:47:07

纬领网络 · 2025-06-23 22:47:07 发布

在linux系统ubuntu系统里安装的GPU的pytorch测试GPU时提示如下错误信息

/home/haiyc/miniconda3/lib/python3.13/site-packages/torch/cuda/__init__.py:174: UserWarning: CUDA initialization: CUDA driver initialization failed, you might not have a CUDA gpu. (Triggered internally at /pytorch/c10/cuda/CUDAFunctions.cpp:109.)
return torch._C._cuda_getDeviceCount() > 0
PyTorch CUDA状态: False

解决方法：

CUDA_DEVICE_ORDER="PCI_BUS_ID"

PYTORCH_NVML_BASED_CUDA_CHECK=1

CUDA_VISIBLE_DEVICES=0,1,2,3

python -c "import torch;print(torch.cuda.is_available());"