ubuntu16.04下多种CUDA版本的安装和切换

       因为配置不同的项目,不同的项目使用的tensorflow、pytorch的版本不同,对应的CUDA版本也不同,记录一下,多种版本CUDA的安装和切换。

       从图片可见,自己配置了三个版本的CUDA。

 

       下面记录安装和配置过程。

     (1)从官网下载所需要版本的CUDA:https://developer.nvidia.com/cuda-toolkit-archive

       文件类型为:runfile。

     (2)安装CUDA

        cd到刚下载的文件目录

        sudo sh cuda_10.0.130_410.48_linux.run

        一直按回车键到100%,输入accept,后续输入指令选择时一定要看清楚,视情况决定。

        最关键的是问是否要安装驱动的时候:install NVIDIA Accelerated Graphics Driver for Linux-x86_64,输入n(no)。因为已安装其他版本的CUDA,是已经配置好驱动的。

        问题是否建立一个cuda的软链接:Do you want to install a symbolic link at /usr/local/cuda,输入n,后续我们切换时自己建立软链接。

        其他视情况决定,比如一些文件的默认路径等问题。

      (3)设置环境变量

        在~/.bashrc文件中添加环境变量,如果有配置好的CUDA,这步应该已经完成,如果没有,通过几个export操作配置,配置完后source ~/.bashrc 

      (4)cudnn的安装

        官网下载CUDA对应版本的cudnn:https://developer.nvidia.com/rdp/cudnn-archive 

        解压完成后,可以看到cuda文件夹,在当前目录打开终端,执行下面命令:

        sudo cp cuda/include/cudnn.h /usr/local/cuda-10.0/include/

        sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.0/lib64/

        sudo chmod a+r /usr/local/cuda-10.0/include/cudnn.h

        sudo chmod a+r /usr/local/cuda-10.0/lib64/libcudnn*

     (5)建立指向CUDA-10.0的软链接

        sudo rm -rf cuda

        sudo ln -s /usr/local/cuda-10.0 /usr/local/cuda

     (6)查看此时的CUDA版本和cudnn版本

        CUDA版本:

        cat /usr/local/cuda/version.txt  

        nvcc -V

        cudnn版本:

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

        没有问题则说明配置成功,如果要切换不同的CUDA版本,只需要执行第五步即可。

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值