ubuntu14.04安装更新cuda10.0.130,及对应版本的NVIDIA驱动和cudnn

	自己用cuda8.0安装了pytorch,然后发现服务器的cuda版本为10.0.130,二者不匹配(安装版本与服务器版本不一致),使用时出现:torch.cuda.is_avaliable()=False,因此需要更新安装cuda版本,使其与服务器版本一致,再重新安装pytorch。同时需要注意NVIDIA驱动也要更新,否则无法正常使用。

一、关闭X文件
NVIDIA驱动的更新要求关闭X文件,操作如下:

  1. 进入管理员模式
    重新打开一个纯命令窗口,切换到管理员账号下:su 账号:XXX 密码:XXX
  2. 关闭X锁文件和文件夹
    sudo init 3 # 注意空格
    rm /tmp/.X*
    rm -rf /tmp/.X*
  3. 重启服务器
    reboot

二、更新cuda版本10.0.130,保留cuda8.0,建立软连接

  1. 下载安装包
    cuda下载地址
    选择与自己系统对应版本的安装包 在这里插入图片描述
  2. 安装
    mkdir cuda_10.0 #用于放置cuda10.0安装文件 和Samples
    sudo sh cuda_10.0.130_410.48_linux.run
    会出现很多提示信息,可以按空格键跳过,直到more达到100%
    安装选项:
    记得在安装驱动和软连接的地方选择yes!!!
    安装驱动选项选择:yes为避免删除以前安装的其它版本,可通过建立软连接灵活使用不同版本
    安装完成
  3. 修改配置文件
    vim ~/.bashrc
    键盘输入G #vim一般模式转为命令模式,光标至文件最后一行首字母处
    键盘输入I #换成插入模式,修改文件
    在文件最后一行添加:
    在这里插入图片描述
    键盘按ESC #从命令模式回到一般模式
    命令行输入:wq #保存文件修改并退出
    source ~/.bashrc #更新配置文件
  4. 建立软连接
    sudo ln -s /usr/local/cuda-10.0 /usr/local/cuda
    若要修改软连接:(如与cuda8.0连接)
    sudo rm -rf /usr/local/cuda
    sudo ln -s /usr/local/cuda-8.0 /usr/local/cuda #一定要先删除现有连接,再重新连接
  5. 查看版本
    查看cuda版本:nvcc –version
    查看驱动版本:nvidia-smi
    (若左右版本号一致,则驱动更新成功)
    若左右版本号一致,则驱动更新成功

三、安装cudnn

  1. 下载安装包
    cudnn下载地址
    在这里插入图片描述
  2. 解压文件
    cp cudnn-8.0-linux-x64-v5.1.solitairetheme8 cudnn-8.0-linux-x64-v5.1.tgz
    tar -xvf cudnn-8.0-linux-x64-v5.1.tgz
  3. 更新cuda中的cudnn文件
    #将cudnn解压后的文件夹cuda中的文件拷贝至刚刚安装的/usr/local/cuda下
    sudo cp cuda/include/.h /usr/local/cuda/include/
    sudo cp cuda/lib64/lib
    /usr/local/cuda/lib64/
  4. 创建软连接
    cd /usr/local/cuda/lib64/
    sudo rm -rf libcudnn.so libcudnn.so.7 #删除
    sudo ln -s libcudnn.so.7.6.0 libcudnn.so.7 #链接
    sudo ln -s libcudnn.so.7 libcudnn.so
    sudo ldconfig
  5. 查看cudnn版本
    cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值