前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >点云深度学习环境配置指南(一)Ubuntu16.04+RTX2080ti

点云深度学习环境配置指南(一)Ubuntu16.04+RTX2080ti

作者头像
点云乐课堂
发布2020-05-18 15:41:05
1.2K0
发布2020-05-18 15:41:05
举报
文章被收录于专栏:3D点云深度学习3D点云深度学习

【今日导读】想做点云深度学习?先把环境配置好吧。本期为初学者带来环境配置指南,有需求的同学赶快上手吧。配置为:

Ubuntu16.04+RTX2080ti+NVIDIA驱动+cuda10.1+cudnn7.5。


一、安装NVIDIA驱动

法一(优先推荐):

先添加NVIDIA 的ppa 源:

代码语言:javascript
复制
$ sudo add-apt-repository ppa:graphics-drivers/ppa
$ sudo apt-get update

打开system settings -> Softwares & Updates -> Additional Drivers,或者在中终端输入:

代码语言:javascript
复制
$ software-properties-gtk

以上是可以使用的驱动,默认会使用Nouveau。可以打开终端,查看为你的显卡推荐使用的驱动,输入:

代码语言:javascript
复制
$ sudo ubuntu-drivers devices

这里给我推荐使用的是nvidia-387,要安装可以在终端输入:

代码语言:javascript
复制
$ 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)、删除原有驱动(可选):

代码语言:javascript
复制
$ sudo apt-get remove --purge nvidia*

(2)、禁用nouveau,安装NVIDIA需要把系统自带的驱动禁用

打开文件:

代码语言:javascript
复制
sudo gedit /etc/modprobe.d/blacklist.conf

在文本最后添加以下内容:

代码语言:javascript
复制
blacklist nouveau
option nouveau modeset=0  

然后保存退出

执行:

代码语言:javascript
复制
sudo update-initramfs -u

重启电脑,命令行输入:

代码语言:javascript
复制
lsmod | grep nouveau

没有任何输出的话就代表禁用成功。

3.安装NVIDIA驱动程序(准备另一台电脑,用来看教程或者是查找需要输入的命令)

禁用图形界面服务:

代码语言:javascript
复制
sudo service lightdm stop

按Ctrl+Alt+F1进入命令行,输入用户名、密码登录(通过Ctrl+Alt+F7可返回界面)

下载的是.run文件,首先进入到下载目录:

代码语言:javascript
复制
cd Downloads

(这里应该写你当时把.run文件下载到的文件夹的路径,一般默认是下载文件夹,如果你在这里出现错误,比如找不到文件,或者路径是中文的,可以尝试把.run文件移动到一个home下,这样可以直接打:cd /home/你的用户名/ 这样就可以打开了)

然后输入:

代码语言:javascript
复制
sudo chmod a+x NVIDIA-Linux-x86_64-418.56.run 

(根据自己下载的文件修改文件名)这是不会有什么输出。

接着安装:

代码语言:javascript
复制
sudo ./NVIDIA-Linux-x86_64-410.66.run -no-opengl-files

参数:

代码语言:javascript
复制
(–no-opengl-files 只安装驱动文件,不安装OpenGL文件。这个参数最重要
–no-x-check 安装驱动时不检查X服务
–no-nouveau-check 安装驱动时不检查nouveau

后面两个参数可不加。)

接着就是一直选择它默认的那个选项就好。安装完成后

代码语言:javascript
复制
$ 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、切换到相应目录,在终端中运行

代码语言:javascript
复制
$ 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 文件:

代码语言:javascript
复制
$ sudo gedit ~/.bashrc

打开文件后将下面三行内容加入进去:

代码语言:javascript
复制
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

保存并退出,输入命令:

代码语言:javascript
复制
$ source ~/.bashrc

4、测试是否安装成功,终端依次输入下面三行命令:

代码语言:javascript
复制
$ 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”。

下载完成后解压并进入文件夹,依次输入一下命令:

代码语言:javascript
复制
$ 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版本:

代码语言:javascript
复制
$ cat/usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

Ubuntu16.04+RTX2080ti上的NVIDIA驱动+cuda10.1+cudnn7.5安装完毕!!

本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2019-04-07,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 3D点云深度学习 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • Ubuntu16.04+RTX2080ti上的NVIDIA驱动+cuda10.1+cudnn7.5安装完毕!!
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档