欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Ubuntu18.04 + cuda 10.2 + cudnn 配置

程序员文章站 2022-07-12 09:36:36
...

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. 注意 此时要加入环境变量

    #终端打开./bashrc
    
    sudo 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,还有很多不会,也不会截图,所以写的比较烂,之后一点点补充把。有问题,评论一起讨论。

相关标签: linux ubuntu