tensorflow gpu 的安装

1、安装nvidia-384
安装完系统之后升级一下系统
sudo apt-get update
sudo apt-get upgrade
ubuntu的官方apt源就有最新最稳定的nv显卡驱动。使用ubuntu自己带的驱动安装工具直接安装就可以。
ubuntu-drivers devices

sudo apt-get install nvidia-384

备注也可以使用下载.run等方式安装,因电脑而异,我的电脑安装的时候会出现各自各样的问题所以采取了简单的方法
提供俩篇文章希望对安装的过程有帮助
[第一篇链接]该篇文章解决了循环登录、重启黑屏、驱动不工作(https://www.cnblogs.com/matthewli/p/6715553.html)
第二篇
检验 驱动是否正确安装
cat /proc/driver/nvidia/version
检验驱动是否有效工作
nvidia-smi #若列出GPU的信息列表,表示驱动安装成功
tensorflow gpu 的安装_第1张图片
nvidia-settings #若弹出设置对话框,亦表示驱动安装成功

tensorflow gpu 的安装_第2张图片

2、 安装cuda
a、下载
cuda的下载网址:https://developer.nvidia.com/cuda-downloads?target_os=Linux
sudo dpkg -i cuda-repo-ubuntu1604-9-1-local_9.1.85-1_amd64.deb
sudo apt-key add /var/cuda-repo-9-1-local/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda

b、写入环境变量
gedit ~/.bashrc
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=:/usr/local/cuda/lib64
c、source
source ~/.bashrc

3、安装cudnn
a、下载网址https://developer.nvidia.com/cudnn
b、install
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

*4、Installing Bazel*
参考:https://docs.bazel.build/versions/master/install-ubuntu.html
sudo apt-get install openjdk-8-jdk
sudo add-apt-repository ppa:webupd8team/java
sudo apt-get update && sudo apt-get install oracle-java8-installer

echo "deb [arch=amd64] http://storage.googleapis.com/bazel-apt stable jdk1.8" | sudo tee /etc/apt/sources.list.d/bazel.list

curl https://bazel.build/bazel-release.pub.gpg | sudo apt-key add -

a、下载bazel
链接:https://github.com/bazelbuild/bazel/releases
特别慢,后续我会给一个我下载好的一一方便后来的使用
b、install
chmod +x bazel--installer-linux-x86_64.sh
./bazel--installer-linux-x86_64.sh –user

c、export
export PATH=” PATH: P A T H : HOME/bin”

5、安装tensorflow
a、
将tensorflow放到/home下
执行./configue
配置configure,不懂的就按回车选择默认方式
b、执行如下命令,在/home/tensorflow
记得加sudo或者在root用户下执行
命令1、
bazel build -c opt //tensorflow/tools/pip_package:build_pip_package

命令2、
bazel build -c opt –config=cuda //tensorflow/tools/pip_package:build_pip_package

命令3、
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
这时在/tmp/tensorflow_pkg/会有一个*.mkl生成
例如:tensorflow-1.5.0rc0-cp27-cp27mu-linux_x86_64.whl

命令4、pip 安装该包
sudo pip install tensorflow-1.5.0rc0-cp27-cp27mu-linux_x86_64.whl

你可能感兴趣的:(ubuntu-配置,ubuntu,tensorflow)