Ubuntu18.04 + cuda 10.2 + cudnn 配置

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

  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值