工具使用-Ubuntu16.04下cuda和cudnn的卸载,更新和多版本切换

发布 : 2019-06-13 浏览 :

cuda安装

  • 首先去Nvidia官网选择下载自己希望安装的cuda版本。

    download

    服务器上直接下载, 可以右键复制下载链接,然后使用如下指令

    1
    wget https://developer.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.168_418.67_linux.run
  • 安装cuda

    1
    sudo sh cuda_10.1.168_418.67_linux.run
  • 安装过程中一直enter键,直到协议结束出现下面的选择

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    Do you accept the previously read EULA?
    accept/decline/quit: accept # 同意协议

    Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 396.37?
    (y)es/(n)o/(q)uit: n # 如果显卡驱动没安装,则需要选择yes, 这里是安装新的版本,所以旧的cuda安装时,已经安装过显卡驱动,所以可以选择no, 不再安装显卡驱动

    Install the CUDA 9.2 Toolkit?
    (y)es/(n)o/(q)uit: y # 安装工具包

    Enter Toolkit Location
    [ default is /usr/local/cuda-9.2 ]: # 这里就选择默认安装路径就好

    Do you want to install a symbolic link at /usr/local/cuda?
    (y)es/(n)o/(q)uit: y # 是否创建软连接, 这个软连接其实就可以用来选择不同的cuda版本,因为bashrc中的cuda路径都是使用该软连接指定的。

    Install the CUDA 9.2 Samples?
    (y)es/(n)o/(q)uit: y # 安装样本代码

    至此cuda10安装成功。

cuda版本切换

  • 不同版本的cuda都是安装在默认路径/usr/local下的, 而在该路径下还有一个软连接/usr/local/cuda

softlink

判断当前软连接链接的是哪个版本为cuda:

whichcuda

可以看到当前软连接采用的是cuda9.0。

如果想切换成cuda8.0, 直接修改软连接则可

1
sudo ln -s /usr/local/cuda /usr/local/cuda8.0
  • 当然,这么做的前提是配置文件的路径需要设置好

打开~/.bashrc文件,修改环境变量, 如果没有则添加

1
2
3
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64

每次修改~/.bashrc文件之后都需要重新激活一次。

1
source ~/.bashrc
  • 查看当前cuda的版本

    1
    nvcc --version

    nvcc_version

cudnn 安装

选择cuda版本一致的cudnn版本下载。

cudnn_download

下载下来是soliteairetheme8文件, 转换成tgz文件之后进行解压。

1
2
cp cudnn-9.2-linux-x64-v7.4.2.24.solitairetheme8 cudnn-9.2-linux-x64-v7.4.2.24.tgz
tar -xvzf cudnn-9.2-linux-x64-v7.4.2.24.tgz

然后将里面相应文件拷贝到新安装的cuda中

1
2
sudo cp include/cudnn.h /usr/local/cuda9.0/include/
sudo cp lib64/lib* /usr/local/cuda9.0/lib64/

至此cudnn安装完成。

cuda卸载

1
2
sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl
sudo rm -rf /usr/local/cuda-8.0
本文作者 : zhouzongwei
原文链接 : http://yoursite.com/2019/06/13/cuda-update/
版权声明 : 本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明出处!

知识 & 情怀 | 赏或者不赏,我都在这,不声不响

微信扫一扫, 以资鼓励

微信扫一扫, 以资鼓励

支付宝扫一扫, 再接再厉

支付宝扫一扫, 再接再厉

留下足迹