Centos7.9gpu版本pytorch安装过程(仍有问题,求助)

在Centos7.9上安装GPU版PyTorch时,尽管能检测到NVIDIA驱动版本,但无法正常使用。过程中涉及到CUDA、cuDNN的卸载与安装,以及GCC的升级。在安装NVIDIA驱动时出现错误,提示内核模块已加载,导致安装失败。已尝试重新安装,但问题未解决,寻求帮助。
摘要由CSDN通过智能技术生成

目录

问题:能够查到nvidia驱动版本,但却不可用

(一)准备工作:

卸载原有的Pytorch

卸载cuda  cudnn

查看nvidia-smi

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

查看GCC版本4.8.5

gcc与cuda版本对应关系

(二)开始安装

升级gcc至8.3.1

安装cuda

安装cudnn

安装GPU版的pytorch

安装NVIDIA


问题:能够查到nvidia驱动版本,但却不可用

 

解决步骤:

  1. (一)准备工作:

  • 卸载原有的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

参考

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值