ubuntu安装TensorFlow(支持不同版本)

  • Post author:
  • Post category:其他



CUDA+CUDNN准备:


1.安装NVIDIA驱动

(1)查询NVIDIA驱动

首先去官网(http://www.nvidia.com/Download/index.aspx?lang=en-us)查看适合自己显卡的驱动(下载runfile文件):

2)安装驱动

先按Ctrl + Alt + F1到控制台,关闭当前图形环境

  $sudo service lightdm stop

卸载可能存在的旧版本 nvidia 驱动

sudo apt-get remove --purge nvidia*
安装驱动可能需要的依赖 
$sudo apt-get update
$sudo apt-get install dkms build-essential linux-headers-generic
 sudo nano /etc/modprobe.d/blacklist-nouveau.conf

 在文件 blacklist-nouveau.conf 中加入如下内容:
blacklist nouveau
options nouveau modeset=0
保存退出,执行
 sudo update-initramfs -u

ctrl+O保存。然后回车 ctrl+X 退出
reboot重启

运行驱动安装文件

重启后再次进入字符终端界面(Ctrl + Alt + F1),并关闭图形界面

  $sudo service lightdm stop

安装驱动

  $sudo chmod u+x NVIDIA-Linux-x86_64-361.45.11.run

  $sudo ./NVIDIA-Linux-x86_64-361.45.11.run(sudo ./NVIDIA-Linux-x86_64-375.20.run –no-opengl-files)
或 sudo ./NVIDIA-Linux-x86_64-375.20.run –no-x-check –no-nouveau-check –no-opengl-files
kenel 不装选Yes; X-config Yes

最后重新启动图形环境 
$sudo service lightdm start

可以通过以下命令确认驱动是否正确安装

$cat /proc/driver/nvidia/version


2.安装cuda8.0

9.0的时候用runfile没有安装成功,用安装包装的,ok了。安装包本地:

`sudo dpkg -i cuda-repo-ubuntu1604-9-2-local_9.2.88-1_amd64.deb`
`sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub`
`sudo apt-get update`
`sudo apt-get install cuda`

可能出现 Driver/library version mismatch 的问题,重启,或者按照

此方法


需要进入安全模式安装(lightdm)

,在安装cuda时是可以自动安装NVIDIA驱动的,但是这样安装的驱动版本较低,且可能有问题,所以需要先安装NVIDIA驱动,在安装CUDA时选择不安装驱动。

切换到

cuda_8.0.44_linux.run

的目录,执行

$ sudo sh cuda_8.0.44_linux.run
 


安装CudNN

下在对应的版本(cudnn-9.2-linux-x64-v7.1)按照官网教程,解压复制:

tar -xzvf cudnn-9.0-linux-x64-v7.tgz

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

/usr/local/cuda/lib64/libcudnn*


正式安装:


一、查看系统所安装的python版本

打开终端输入指令:python,如图1所示,我的系统是ubuntu16.04.03,默认安装的python版本为2.7.12。




二、安装python对应版本的pip和依赖包

若python版本为2.7,则输入如下命令:

sudo apt-get install python-pip python-dev

若python版本为3.x,则输入如下命令:

sudo apt-get install python3-pip python3-dev


三、升级pip版本

在装tensorflow之前,不管是不是最新的pip版本,都要更新一下,具体命令如下:

python 2.7版本:sudo pip install –upgrade pip

python 3.x版本:sudo pip3 install –upgrade pip

四、更改pip源地址(提高下载速度)

修改 ~/.pip/pip.conf (没有就创建一个文件夹及文件,文件夹要加”.”,表示是隐藏文件夹),内容如下:

[global]

index-url = https://pypi.tuna.tsinghua.edu.cn/simple

[install]

trusted-host=mirrors.aliyun.com


五、安装TensorFlow

TensorFlow可以安装CPU和GPU两种版本,CPU版本安装命令如下:

python 2.7版本:sudo pip install tensorflow

python 3.x版本:sudo pip3 install tensorflow

GPU版本安装命令如下:

python 2.7版本:sudo pip install tensorflow-gpu

python 3.x版本:sudo pip3 install tensorflow-gpu

若上述命令执行过程没有报错,则安装成功。


六、测试安装结果

进入python编译环境,导入TensorFlow,做一个简单的加法运算,如图2所示。



版权声明:本文为intflojx原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。