目录
问题:能够查到nvidia驱动版本,但却不可用
解决步骤:
-
(一)准备工作:
-
卸载原有的Pytorch
conda uninstall pytorch
conda uninstall libtorch
-
卸载cuda cudnn
cd到 /usr/local/cuda-10.1/bin下,执行 ./cuda-uninstaller
cd /usr/local/cuda-10.1/bin
./cuda-uninstaller
-
查看nvidia-smi

-
官方cuda与nvidia的linux版本对应关系

-
查看GCC版本4.8.5

-
gcc与cuda版本对应关系
Centos 7默认gcc版本为4.8,有时需要更高版本的,这里以升级至8.3.1版本
(二)开始安装
-
升级gcc至8.3.1
参考

在Centos7.9上安装GPU版PyTorch时,尽管能检测到NVIDIA驱动版本,但无法正常使用。过程中涉及到CUDA、cuDNN的卸载与安装,以及GCC的升级。在安装NVIDIA驱动时出现错误,提示内核模块已加载,导致安装失败。已尝试重新安装,但问题未解决,寻求帮助。
最低0.47元/天 解锁文章
4500

被折叠的 条评论
为什么被折叠?



