当前位置: 代码迷 >> 综合 >> Ubuntu18.04 + cuda 10.2 + cudnn 配置
  详细解决方案

Ubuntu18.04 + cuda 10.2 + cudnn 配置

热度:99   发布时间:2023-11-13 23:29:39.0

Ubuntu18.04 + cuda 10.2其他 + cudnn 配置

  • NVIDIA驱动安装
    • 直接采用命令行的安装形式
  • CUDA 10.x 安装
  • Cudnn 安装
  • 接下来会安装 anaconda pytorch ,一点点来,目前刚用ubuntu,还有很多不会,也不会截图,所以写的比较烂,之后一点点补充把。有问题,评论一起讨论。

NVIDIA驱动安装

直接采用命令行的安装形式

建议直接采用命令行的安装,这样省事省力。建议开始安装都试一下这种方式

  1. 打开软件更新。
    我之前关闭了软件设置更新,导致安装不成功。所以建议大家先更新以下系统。
  2. 安装驱动
    这里就不写如何看推荐版本了,直接安装把。
    // 默认安装最新满足要求的
    sudo ubuntu-drivers autoinstall
    
  3. 如果安装失败了,可以百度艘以下,apk等安装方式,比较麻烦。

最终安装后,进入 系统设置->详细信息 能看到 图形 栏出现了自己电脑的显卡。

CUDA 10.x 安装

  1. 安装好驱动后,可以在终端运行 nvidia-smi 最上面有个CUDA Version : xxx ,这个一般是支持CUDA 的最高版本。 一般用9.1, 10.1,10.2 目前挺普遍的。(2021年)

  2. 进入CUDA官网 link.,这里有一些问题,就是进取默认最新的版本,恨fan,想装想要的版本建议 直接在百度搜索一下。

  3. 网可以的话,建议 runfile 安装方式

  4. 等待结果就可以拉。 如果不成功,就还是下载再安装(麻烦)。

  5. 注意 此时要加入环境变量

    #终端打开./bashrcsudo gedit ~/.bashrc 
    

    // 安装后运行
    在 ./bashrc文件中最后加入: 一定要进目录看看文件名字,我装得10.2 所以文件名字cuda-10.2

    export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
    export CUDA_HOME=/usr/local/cuda-10.2 
    

    命令行中输入 : source ~/.bashrc
    命令行中输入 : nvcc -V 会有结果。

Cudnn 安装

比较麻烦,但是网上资料很多。

  1. 进入官网link,登陆(注册以下,多等一会就好了)。
  2. 下载对应的版本 的(cuDNN Library for Linux (x86))。
  3. 下载完成后解压并进入文件夹:
    // 将CUDNN 复制到CUDA中
    sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ 
    sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ 
    sudo chmod a+r /usr/local/cuda/include/cudnn.h 
    sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
    sudo cp cuda/include/cudnn_version.h /usr/local/cuda/include/
    
  4. 检查是否安装成功
    cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2 
    

有结果就安装成功了。

接下来会安装 anaconda pytorch ,一点点来,目前刚用ubuntu,还有很多不会,也不会截图,所以写的比较烂,之后一点点补充把。有问题,评论一起讨论。