问题描述

运行代码报错：
RuntimeError: The NVIDIA driver on your system is too old (found version 11040). Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx Alternatively, go to: https://pytorch.org to install a PyTorch version that has been compiled with your version of the CUDA driver.

原因

项目环境中的pytorch版本与系统的nvidia显卡驱动版本不匹配。

1.查看pytorch版本

python -c "import torch; print(torch.__version__)"

# 2.4.0+cu121

2.查看CUDA版本信息

nvidia-smi

# 11.4 

解决方案

安装 PyTorch 1.11.0 版本，并且使用 CUDA 11.3

pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html