使用 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=640x480
→GRUB_GFXMODE=1920x1080
$ sudo update-grub
1.安装Nvidia驱动,Cuda,Cudnn
-
驱动安装:
驱动使用ubuntu自带的 Software Updater > Settings > Additional
Drivers,选择要安装的驱动版本,Apply Changes。 -
Cuda:
cuda推荐使用“deb(local)”的安装方式进行安装,只需依次执行下方生成的命令即可。
其中:第三句 wget 的下载可能较慢,可以通过其他方式下载到本地,之后跳过此行执行。 -
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