tensorflow的安装和注意事项

想了一下还是把tensorflow安装的过程整理一下吧，万一时间久了忘了呢。

终于tensorflow的安装可以告一段落了，内心还是很兴奋的，这次还是好好的整理下。

尤其是注意的地方，往往时我折腾了好久，查阅了大量的资料，测试了好多次，才验证出来的硕果。

1、准备工作

　　1、更换源，好的软件源，直接决定你的安装速度。这里选择清华的。

　　操作：进入：设置 --> 软件和更新 --> Ubuntu软件，打开下载自：选择其他，选择清华。点击选择服务器确认，在弹出框选择，重新载入。

　　2、更新软件包，防止有些工具没有安装上。

sudo apt update

　　3、打开：设置 --> 软件和更新 --> 附加驱动

　　选择NVIDIA公司专有的最新驱动，选择应用更改。时间较长，请耐心等待。

2、安装cuda

　　下载文件，这里我选择run本地文件，进入文件所在目录，并运行，XXX代指文件名，历史版本

sudo sh XXX.run

　　安装注意，这时会出现很多信息，按ctr+c关闭信息，再按accept接受。

　　接下来询问是否，安装驱动，输入n，不要安装。因为已经安装过。

　　一路输入y，或者回车。

　　进入/home/用户名/NVIDIA_CUDA-8.0_Samples/1_Utilities/deviceQuery，打开终端，输入

sudo make

　　然后输入

./deviceQuery

　　如果显示pass，说明cuda安装成功，如果显示说，cuda驱动不满足于cuda运行库，说明驱动太旧，或者cuda太新。参照修改。

3、下载并安装cudnn（本段参照于http://blog.csdn.net/m0_37924639/article/details/78785699）

　　下载cudnn，安装CUDNN

　　进入CUDNN安装包所在目录，执行以下命令：

sudo dpkg -i runtime包.deb
sudo dpkg -i developer包.deb
sudo dpkg -i 代码sample包.deb

　　报错查看此项https://blog.csdn.net/u011636567/article/details/77162217

　　至此，CUDNN安装完成。

　　验证CUDA和CUDNN是否安装成功

　　CUDNN的code sample可以用来检查CUDNN和CUDA是否安装成功，执行以下命令：

sudo cp -r /usr/src/cudnn_samples_v7/ $HOME
cd $HOME/cudnn_samples_v7/mnistCUDNN
sudo make clean
sudo make
sudo ./mnistCUDNN

　　正常情况下执行以上代码会得到Test passed!的结果。如果在make步出错，那么可能gcc需要降级（由于一直未出现类似错误，作者未验证）；

　　如果出现CUDA driver version is insufficient for CUDA runtime version，说明你的显卡驱动版本低于cuda工具包的版本。

　　如果下载的是压缩包，那个运行库，可以使用下面的。

cd  /home/***(自己的用户名)/Desktop/###(这个命令意思是找到刚刚我们用U盘传过来的文件)
tar xvzf cudnn-8.0-linux-x64-v5.1-ga.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*

4、版本说明

　　1、显卡驱动要高于cuda版本，cudnn要选择和cuda版本一样的

　　2、建议安装的cuda版本和驱动的版本一致，比如我的驱动目前只能是384.11（9.0版本），最好安装cuda工具包9.0

　　3、安装对应的cudnn9.0

4、安装tensorflow　　

在配置完cudnn之后，可以直接在终端输入

sudo pip install tensorflow==1.5.0

或者安装CPU版本

sudo pip install tensorflow==1.5.0

如果网速不好的话，也可以自己在https://github.com/tensorflow/tensorflow下载whl文件，往下翻，在installation下面，然后使用

sudo pip install xxx.whl

　5、简便安装的方法

如果确定了自己的显卡支持的最高版本驱动，可以直接参照下面的方法：

1、cuda的下载和安装

　　下载适合你的cuda版本（包含驱动，即驱动版本）

　　关闭目前的图形界面，

sudo service lightdm stop

　　按住ctr + alt + f1,进入命令行界面，输入帐号和密码，并且登录。

　　进入你的cuda下载目录，比如我的在Download里面

cd Download
sudo sh XXX.run --silent --overide

　　不输入参数，先accept，一路y或者回车默认

　　打开图像界面

sudo service lightdm start

　　进入图形界面，按下ctr + alt + f7

　　进入用户主路径

cd ~

　　进入cuda的例子make并且运行，验证是否正确安转cuda

cd /home/shuailing/NVIDIA_CUDA-9.0_Samples/1_Utilities/deviceQuery
make
./dedeviceQuery

　　如果最后一行显示pass，说明安装成功。

　　注意：这个方法，我在工作站安装成功，在自己的电脑上失败，出现的现象时，反复登录。

　　解决方法：先安装显卡驱动，在安装cuda，在安装cuda的时候，询问是否安装cuda驱动（显卡驱动），选择否。

　　如果，出现了重启可以切换回之前的驱动。

2、下载并安装cudnn

　　下载cudnn，安装CUDNN

　　进入CUDNN安装包所在目录，执行以下命令：

sudo dpkg -i runtime包.deb
sudo dpkg -i developer包.deb
sudo dpkg -i Codesample包.deb

注意：这里需要按照这个顺序来进行，负责最后的代码测试包，是无法安装的。

　　如果测试说，找不到，libdart文件，说明没有把 cuda的lib库加入LD_LIBRARY_PATH中。

　　这个之前也加入过，但是总是有其他的意外，这里把加入环境变量，再说一便。

sudu gedit .bashrc

　　把下面两行加入.bashrc最后

export PATH=$PATH:/usr/local/cuda-9.0/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64

　　输入下面命令使其立即生效，之前的cudnn测试的窗口。需要重新打开，才能更新新的环境变量。

source .bashrc

　　如果是解压放入到库中的，还需要对其libcudnn进行链接，ln -s 源文件目标文件

3、安装tensorflow-gpu

　　下载tensorflow-gpu，之后进入下载路径

　　pip安装

sudo pip install tensorflow_gpu-1.6.0-cp27-cp27mu-manylinux1_x86_64.whl

这个tensorflow-gpu==1.6.0，需要实测，cuda9+cudnn7.0可以用，之前用cuda9+cudnn7.1运行程序说，找到cudnn7.1版本，但是期望是7.0。程序报错。

4、验证

python
import tensorflow

　　不报错说明安装成功。　　

6、源码安装

1、克隆tensorflow

git clone https://github.com/tensorflow/tensorflow

2、进入下载目录的tensorflow主目录，配置编译环境

./configure

　　除了Cuda、Cudnn相关的yes，其余的不相关的都选择n。剩下的默认回车就好。

3、完成之后编译

sudo bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
sudo mkdir /tmp/tensorflow_pkg
sudo bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
sudo pip install/tmp/tensorflow_pkg/tensorflow-1.2.0rc2-cp27-cp27mu-linux_x86_64.whl

　　之前因为权限不够，编译出错不加载，执行第三条说找不到文件。现在全部加了权限。

4、参照上面的测试

5、最后的注意：

　　描述：在我运行程序的时候，终端运行正常使用GPU，可是在pycharm中运行就报错，说找不到libcublas.so.9.0.源码编译的时候也说找不到/usr/bin/ld的什么文件。

　　尝试的方法：tensorflow降级，结果不可以。实际上我的cuda和cudnn都是要求的9.0和7.0,明明在/usr/local/cuda/lib64中可以找到这个文件的。可是不论是降级还是配置cudnn的libcudnn.so的链接都不行。

　　后来，在手机上偶然查到。

　　解决方案：在/etc/ld.so.conf中最后一行添加/usr/local/cuda-9.0/lib64。或者在ld.so.conf.d这个目录新建一个档案，以.conf结尾（建议），最后输入sudo ldconfig使其立即生效。

　　解释：其实上面那个含有libculas的路径，正常的已经被添加到LD_LIBRARY_PATH中了，可是程序运行的时，不一定回读到这个变量，所以需要设置/etc/ld.so.conf把，含有libcublas的路进加入的系统中。

本站文章如无特殊说明，均为本站原创，如若转载，请注明出处：tensorflow的安装和注意事项 - Python技术站