1、安装nvidia显卡驱动

安装深度学习框架需要使用cuda/cudnn(GPU)来加速计算,而安装cuda/cudnn,首先需要安装nvidia的显卡驱动。
http://www.geforce.cn/drivers  或者  http://www.nvidia.com/Download/index.aspx?lang=en-us
ubuntu 16.04默认安装了第三方开源的驱动程序nouveau,安装nvidia显卡驱动首先需要禁用nouveau,不然会碰到冲突的问题,导致无法安装nvidia显卡驱动。

sudo vim /etc/modprobe.d/blacklist.conf   #编辑文件blacklist.conf
在文件最后部分插入以下两行内容
blacklist nouveau
options nouveau modeset=0
sudo update-initramfs -u     #更新系统后,重启系统(一定要重启)

验证nouveau是否已禁用
lsmod | grep nouveau     #没有信息显示,说明nouveau已被禁用,接下来可以安装nvidia的显卡驱动。

(1)、禁用自带的nouveau驱动(重要步骤!)
运行命令创建相关文件:
sudo gedit /etc/modprobe.d/blacklist-nouveau.conf
并在文件中添加如下内容:
blacklist nouveau         options nouveau modeset=0
sudo update-initramfs -u     #更新后,重启系统

lsmod | grep nouveau          #查看是否禁用成功。如如果没有输出则禁用成功

Ctrl+Alt+F1   #通过如下命令进入命令行模式

sudo service lightdm stop   #关闭X-Window  

(2)、安装驱动:
sudo apt-get remove –purge nvidia*    #卸载你电脑中此刻有的nvidia的驱动

sudo add-apt-repository ppa:xorg-edgers/ppa -y
sudo apt-get update
sudo apt-get install nvidia-384       #(根据需要选择版本,注意CUDA9.0与CUDA9.1必须安装387.x的驱动)

sudo service lightdm start     #安装完毕之后重启X-Window

(3)、CTRL+ATL+F1   #进入终端1(输入用户名和密码后和终端很相似,可以大胆进去)
sudo service lightdm stop       #关闭图形界面
sudo apt-get install nvidia-384   #自动下载驱动安装,nvidia-384要和显卡型号匹配)
sudo service lightdm start   #来启动图形界面,并且登录。这个时候应该又回到图形界面。
nvidia-smi      #测试有没安装成功。来看是不是能够输出你的GPU的一些信息。要是不能够输出的话,重启。能够输出的话,也建议重启一次。

2、安装cuda8

https://developer.nvidia.com/cuda-downloads (注意tensorflow 1.6/tensorflow1.7支持CUDA9.0+cuDNN7.0)

sudo apt-get install libcupti-dev   #安装CUDA依赖库

(1)、cd到安装的文件夹,接着执行:sudo sh cuda_8.0.44_linux.run --override

有个让你选择是否安装nvidia驱动时,一定要选择否:  Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 367.48?  因为前面我们已经安装了更加新的nvidia367,所以这里不要选择安装。其余的都直接默认或者选择是即可。 安装完成后,显卡driver版本是9,cuda版本是8。

(2)、配置cuda环境变量  

export PATH="$PATH:/usr/local/cuda-8.0/bin" export 

LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib64"

nvidia-smi  #结果出现以下输出,说明配置成功


3、安装Cudnn

先下载https://developer.nvidia.com/rdp/cudnn-download#a-collapse704-9
简单地说,就是复制几个文件:库文件和头文件。将cudnn的头文件复制到/usr/local/cuda/lib64,将cudnn的库文件复制到/usr/local/cuda/include。  下载下来后,cd 到文件包目录下,解压文件。

tar -zxf cudnn-7.0-linux-x64-v4.0-prod.tgz
cd cuda

#链接到cuda的库里
sudo cp lib64/* /usr/local/cuda/lib64/
sudo cp include/cudnn.h /usr/local/cuda/include/
 
要不要链接cuDNN的库文件:
$ sudo ln -sf /usr/local/lib/libcudnn.so.4.0.7 /usr/local/lib/libcudnn.so.4
$ sudo ln -sf /usr/local/lib/libcudnn.so.4 /usr/local/lib/libcudnn.so
 
#链接完config更新

$ sudo ldconfig

(1)、最后把cudnn的头文件拷贝到/usr/local/cuda-8.0/include下面,库文件拷贝到/usr/local/cuda-8.0/lib64下面
设置环境变量文件.bashrc或profile
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:#LD_LIBRARY_PATH
更新环境变量

souce /etc/profile

4、Anaconda下载地址

安装包大概有500M,下载得一段时间。

https://repo.continuum.io/archive/

https://www.anaconda.com/download/#linux

Ubuntu:Ubuntu下安装Anocuda和Tensorflow最详细攻略Ubuntu:Ubuntu下安装Anocuda和Tensorflow最详细攻略

5、官方Anocuda介绍安装攻略

https://docs.anaconda.com/anaconda/install/linux

lspci | grep -i nvidia   #显示出你的NVIDIA GPU版本信息

gcc --version    #验证是否安装了gcc

sudo apt-get install gcc   #如果没有,可以在终端键入

cd到下载的文件夹,然后输入命令            bash ~/Downloads/Anaconda3-5.1.0-Linux-x86_64.sh             然后点击yes,回车进行安装,接着yes,选择加入环境变量。检查是否安装成功:终端输入python,出现详细信息即安装成功!

conda list    #显示所有的包

(1)、配置anaconda:如果不进行配置,启动spyder等也是不能够运行tensorflow的,因为anaconda中默认anzhaung的spyder等是应用于root的,而非tensorflow。在Terminal中运行命令:
anaconda-navigator #启动anaconda。
在左侧列表中,选择Environment,可看到Anaconda中安装环境为root和tensorflow。选择Tensorflow后,边上会出现一个箭头,这时就可以在右边的列表中查看、安装、卸载Tensorflow下的软件及软件包了。 
安装的软件可在anaconda->home下查看,同样需要选择应用环境。
ipython # terminal下运行python语句,支持自动补全
spyder # 集成开发环境
jupyter # 交互式笔记本

6、进入Jupyter Notebook   

ipython notebook


7、Tensorflow安装、验证

创建Tensorflow环境:conda create -n tensorflow python=3.6

安装tensorflow:  pip install --ignore-installed --upgrade tfBinaryURL

验证是否安装成功:
重启terminal后,输入命令:
source activate tensorflow    #首先**tensorflow环境
python                            #启动python环境
>>> import tensorflow as tf 
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print sess.run(hello)
Hello, TensorFlow!         # 恭喜!安装成功!


相关文章:

  • 2022-01-07
  • 2021-12-18
  • 2021-12-09
  • 2021-12-02
  • 2021-06-20
猜你喜欢
  • 2021-09-12
  • 2021-09-04
  • 2021-09-10
  • 2021-04-22
  • 2021-11-02
  • 2021-11-27
相关资源
相似解决方案