ubuntu18 CUDA版本降级

用别人的代码训练模型时,由于tensorflow-gpu版本太高,报了很多错,一个个改太麻烦,所以直接降级。tensorflow降级为1.14,与此同时也要换成对应版本的CUDA和cuDNN。

1.tensorflow-gpu降级

pip uninstall tensorflow-gpu
pip install tensorflow-gpu==1.14

2.gcc降级

我的服务器本身好像是gcc10.几,现在降到5.5

cd /usr/bin
sudo rm gcc
sudo ln -s gcc-5 gcc
sudo rm g++
sudo ln -s g++-5 g++

查看gcc和g++版本号,均显示gcc version 5.5则成功

gcc -v
g++ -v

3.降级到CUDA9.2

本身是CUDA10,降到CUDA9.2,我没有卸载本来的CUDA,直接在官网下载下面两个文件,cuda_9.2.148.1_linux.run是补丁。

在这里插入图片描述

sudo sh cuda_9.2.148_396.37_linux.run
sudo sh cuda_9.2.148.1_linux.run

直接按q退出,accept,除了安装NVIDIA驱动选项输入n以外(因为之前已经装过显卡驱动了),其余全部选y,默认location都直接回车。
安装完成后,配置环境变量。在.bashrc末尾export一些东西:

sudo gedit ~/.bashrc
export PATH=/usr/local/cuda-9.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.2/lib64:$LD_LIBRARY_PATH
export CUDA_HOME="/usr/local/cuda-9.2:$CUDA_HOME"

然后执行source ~/.bashrc

4.安装cuDNN

在官网下载cuDNN7.4.1,选择library for Linux
在这里插入图片描述
解压:

tar -zxvf cudnn-9.2-linux-x64-v7.4.1.5.tgz

将相关文件复制到刚才安装的cuda中:

sudo cp cuda/include/cudnn.h /usr/local/cuda-9.2/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-9.2/lib64/
sudo chmod a+r /usr/local/cuda-9.2/include/cudnn.h
sudo chmod a+r /usr/local/cuda-9.2/lib64/libcudnn*

5.切换到CUDA9.2

cd usr/local
sudo rm -rf cuda #删除之前创建的软链接
sudo ln -s cuda-9.2 cuda #重建软链接

查看当前的cuda和cuDNN版本:

nvcc -V
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

在这里插入图片描述

  • 0
    点赞
  • 22
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值