【今日导读】想做点云深度学习?先把环境配置好吧。本期为初学者带来环境配置指南,有需求的同学赶快上手吧。配置为:
Ubuntu16.04+RTX2080ti+NVIDIA驱动+cuda10.1+cudnn7.5。
一、安装NVIDIA驱动
法一(优先推荐):
先添加NVIDIA 的ppa 源:
$ sudo add-apt-repository ppa:graphics-drivers/ppa$ sudo apt-get update
打开system settings -> Softwares & Updates -> Additional Drivers,或者在中终端输入:
$ software-properties-gtk
以上是可以使用的驱动,默认会使用Nouveau。可以打开终端,查看为你的显卡推荐使用的驱动,输入:
$ sudo ubuntu-drivers devices
这里给我推荐使用的是nvidia-387,要安装可以在终端输入:
$ sudo apt-get install nvidia-387
安装完成之后,打开Software & Updates,可以看到新安装的显卡驱动已经在使用
再次打开Graphics处显示的就是独立显卡的信息了(由于自己安装的时候忘记截图了,上面几张图是借用别人的,我安装的驱动版本是CUDA版本是418.56,对应的CUDA版本是10.1)
法二:
1.下载驱动文件
去官网下载和自己的显卡适配的驱动文件,是个.run文件。贴个下载地址:
https://www.nvidia.cn/Download/index.aspx
点击“search”之后下载就行,记住它下载到了哪个文件夹里,以后会用的到。在这里搜索到的是最新版本的驱动,如果需要下载其他版本的驱动,点击网页功能栏中的“DRIVES”--->“GEFORCE DRIVES”
点击“START SEARCH”之后搜到多个合适的版本,然后选择自己想要的版本即可。一定记住它下载到了哪个文件夹里,以后会用的到,(一般是保存在Downloads文件夹中)
2.文件已经下载好了,但是在安装之前我我们需要做一些准备。
(1)、删除原有驱动(可选):
$ sudo apt-get remove --purge nvidia*
(2)、禁用nouveau,安装NVIDIA需要把系统自带的驱动禁用
打开文件:
sudo gedit /etc/modprobe.d/blacklist.conf
在文本最后添加以下内容:
blacklist nouveauoption nouveau modeset=0
然后保存退出
执行:
sudo update-initramfs -u
重启电脑,命令行输入:
lsmod | grep nouveau
没有任何输出的话就代表禁用成功。
3.安装NVIDIA驱动程序(准备另一台电脑,用来看教程或者是查找需要输入的命令)
禁用图形界面服务:
sudo service lightdm stop
按Ctrl+Alt+F1进入命令行,输入用户名、密码登录(通过Ctrl+Alt+F7可返回界面)
下载的是.run文件,首先进入到下载目录:
cd Downloads
(这里应该写你当时把.run文件下载到的文件夹的路径,一般默认是下载文件夹,如果你在这里出现错误,比如找不到文件,或者路径是中文的,可以尝试把.run文件移动到一个home下,这样可以直接打:cd /home/你的用户名/ 这样就可以打开了)
然后输入:
sudo chmod a+x NVIDIA-Linux-x86_64-418.56.run
(根据自己下载的文件修改文件名)这是不会有什么输出。
接着安装:
sudo ./NVIDIA-Linux-x86_64-410.66.run -no-opengl-files
参数:
(–no-opengl-files 只安装驱动文件,不安装OpenGL文件。这个参数最重要–no-x-check 安装驱动时不检查X服务–no-nouveau-check 安装驱动时不检查nouveau
后面两个参数可不加。)
接着就是一直选择它默认的那个选项就好。安装完成后
$ service lightdm start #启动图形系统$ sudo reboot #重启计算机:
最后查看是否已经安装好了:
(1)$ nvidia-smi
可以看到驱动版本和建议安装的CUDA版本(图片右上角)
(2)在系统里也能找到NVIDIA的控制面板
(3)并且在系统的setting中的Graphics一栏显示显卡信息
只有上面三个条件同时满足,才算是真正的安装好了驱动。
有时会出现前两个条件都满足,但是系统的setting中的Graphics一栏显示的不是自己的独立显卡信息,而是集显信息。网上有解决办法,在终端输入:
$ nvidia-settings
打开Nvidia X Server Settings,选择PRIME Profile,在此可以自由切换集显和独显。如果没有PRIME Profiles,先reboot。而现实是我的这面板里面没有PRIME Profile选项,即便是reboot计算机,依然没用,所以才强烈推荐第一种安装方法。
二、安装CUDA
1、在https://developer.nvidia.com/cuda-downloads里选择机器环境后下载runfile(local)文件
与搜索驱动是一样的,直接搜索出来的是最新版本,由于驱动和CUDA之间版本之间有一个对应关系,如果不按照下标中给出的信息安装对应版本的CUDA,是装不上的。为了防止小白走弯路,写下查找CUDA版本的操作步骤(点击上图的“Legacy Releases”)
根据自己需要的版本进行查找即可。(例如我需要的是CUDA 10.1,查找后得到下图)
2、切换到相应目录,在终端中运行
$ sudo sh cuda_10.1.105_418.39_linux.run
这个版本的CUDA安装和以前版本的界面不太一样,因此我贴上所有截面图。
(1)输入accept即可
(2)由于之前已经安装过驱动了,现在不用重复安装。将光标移动到Driver一行,按一下键盘的enter键即可排除Driver的重复安装
(3)将光标移动到install行,点击enter键即可安装CUDA
(4)选中“yes”,点击enter
5)安装结束的界面如下
3.加入环境路径
打开.bashrc 文件:
$ sudo gedit ~/.bashrc
打开文件后将下面三行内容加入进去:
export CUDA_HOME=/usr/local/cudaexport PATH=$PATH:$CUDA_HOME/binexport LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
保存并退出,输入命令:
$ source ~/.bashrc
4、测试是否安装成功,终端依次输入下面三行命令:
$ cd /usr/local/cuda/samples/1_Utilities/deviceQuery$ sudo make$ ./deviceQuery
出现下面的结果说明安装成功
三、安装CUDNN
1、到官网在官网下载安装包,需要注册登录才能下载。选择适合自己的版本。贴上网址https://developer.nvidia.com/cudnn
点击“Download cuDNN”,然后需要注册、登录才能下载。登陆后勾选“I Agree To the...”,出现下图,根据自己需要的cuDNN版本进行选择。例如我需要与CUDA10.1对应的7.5.0版本,点击下拉菜单中的“cuDNN Library for Linux”即可以下载。
如果需要其他的版本,可以点击最下方的“Archived cuDNN Releases”。
下载完成后解压并进入文件夹,依次输入一下命令:
$ 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*
在终端查看CUDNN版本:
$ cat/usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
Ubuntu16.04+RTX2080ti上的NVIDIA驱动+cuda10.1+cudnn7.5安装完毕!!!
往期精选:
PointNet++文章及代码
PointCNN原理+代码讲解
frustum-pointnet代码调试bug集锦与解决方法总结
点云深度学习在自动驾驶中的应用——MV3D文章阅读