ubuntu18.04+3090 安装Nvidia驱动、Cuda、Cudnn

使用 3090显卡安装,其他也差不多,只是3090目前只能使用cuda11,而且cuda11.0版本测试安装了两次都有问题,最终安装的目前最新版本cuda11.1.0+cudnn8.0.4
公司网奇慢无比,我都是用的本地安装的方式

安装环境:
ubuntu18.04 Desktop
GeForce 3090
Cuda11.1.0
Cudnn8.0.4
安装日期:2020年11月24日

0. 添加分辨率 调整分辨率至1920x1080

新安装的ubuntu Desktop 18.04只有一个分辨率,1024x768,很难受。
(下文所有的 vim 我自己使用的都是 gedit )

$ sudo vim /etc/default/grub

修改:GRUB_GFXMODE=640x480GRUB_GFXMODE=1920x1080

$ sudo update-grub

1.安装Nvidia驱动,Cuda,Cudnn

  1. 驱动安装:

    驱动使用ubuntu自带的 Software Updater > Settings > Additional
    Drivers,选择要安装的驱动版本,Apply Changes。

  2. Cuda:

    cuda历史版本下载地址

    cudnn下载地址

    cuda推荐使用“deb(local)”的安装方式进行安装,只需依次执行下方生成的命令即可。
    其中:第三句 wget 的下载可能较慢,可以通过其他方式下载到本地,之后跳过此行执行。

  3. Cudnn:
    解压到文件夹

     $ tar -xvf cudnn-***.tgz
    

    将解压缩的文件放入对应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*
    

    将解压缩的文件放入对应cuda项目下

    ps:nvcc -V若提示需要安装Toolkit,可能是由于未配置环境变量,解决方法如下:

    $ sudo vim ~/.bashrc 
    

    添加两句:
    export PATH="/usr/local/cuda-11.1/bin:$PATH"
    export LD_LIBRARY_PATH="/usr/local/cuda-11.1/lib:$LD_LIBRARY_PATH"

    $ source ~/.bashrc
    

验证:

$ nvidia-smi 
$ nvcc -V 

遇到的问题:

1.安装Pytorch之后出现ImportError: libmkl_gnu_thread.so: cannot open shared object file: No such file or directory
  • 首先找到linux上的libmkl_gf_lp64.so,复制当前的路径,很重要!!!
  • 然后 cd /etc/ld.so.conf.d,使用sudo vi runtime-x86_64.conf建立一个文件,将刚刚的路径写入新建的文件中,即,按下i,右键paste,按下Esc,按下:,输入wq,按下回车键。
  • 然后再输入sudo ldconfig完成更新即可。
  • 输入ipython,然后import torch 成功

REF
https://blog.csdn.net/qq_33289694/article/details/103858513

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值