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的信息列表,表示驱动安装成功
具体见图
nvidia-settings #若弹出设置对话框,亦表示驱动安装成功

具体见图

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

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值