美文网首页
Ubuntu 18.04多cuda配置

Ubuntu 18.04多cuda配置

作者: P_Zhi | 来源:发表于2022-09-09 11:24 被阅读0次

    1. cuda下载

    https://developer.nvidia.com/cuda-toolkit-archive

    以11.3为例:

    https://developer.nvidia.com/cuda-11-3-1-download-archive?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=18.04&target_type=runfile_local

    2. cuda安装

    wget https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux.run
    # 给脚本权限
    chmod +x cuda_10.1.105_418.39_linux.run
    # 运行脚本 
    ./cuda_10.1.105_418.39_linux.run --override
    

    安装的时候注意只用选Toolkit就可以,其他的应该不用装,注意不要选软连接。

    此时输入cd /usr/local,可以看到刚装的11.3了。

    3. cuda环境变量配置

    # 首先打开环境配置文件
    sudo gedit ~/.bashrc
    # 在文档的末尾添加这三行,cuda-11.3的地方修改为自己新安装的cuda版本
    export CUDA_HOME=/usr/local/cuda-11.3
    export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64:$LD_LIBRARY_PATH
    export PATH=/usr/local/cuda-11.3/bin:$PATH
    # 生效
    source ~/.bashrc
    

    4. 配套cuDnn

    找到对应cuDnn并下载:

    https://developer.nvidia.com/rdp/cudnn-archive

    本文选择的是cuDnn 8.2.1 for cuda 11.x,只用下载cuDNN Library for Linux (x86_64)就够了。
    下载好后是一个tgz的压缩文件,先解压缩,然后参照下边的代码将对应文件复制到cuda11.3的文件夹中即可。

    tar -zvxf cudnn-10.1-linux-ppc64le-v8.0.5.39.tgz
    sudo chmod 777 -R *
    sudo cp cuda/include/cudnn.h /usr/local/cuda-11.3/include
    sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.3/lib64
    

    总结

    如需其他版本cuda环境,只需重复上述步骤,然后修改~/.bashrc中cuda版本,然后source生效即可。

    本人在服务器上重复上述操作遇到过版本切换失败的情况,此时关闭终端,重新ssh连接一下即可。

    相关文章

      网友评论

          本文标题:Ubuntu 18.04多cuda配置

          本文链接:https://www.haomeiwen.com/subject/uywanrtx.html