针对ECS服务器,使用了GPU进行加速计算,重启后发现CUDA GPUs are available,导致不能运行模型.
-
查看驱动是否工作正常
nvidia-smi -
查看是否安装了驱动
ls /usr/src
- 安装dkms,用来动态加载驱动
#先安装dkms,是ubuntu驱动管理器
sudo apt-get install dkms
#-v 后为版本号
sudo dkms install -m nvidia -v 470.82.01
- 启动显卡驱动
- 查看是否启动成功
解决RuntimeError: No CUDA GPUs are available