Ubuntu16.04安装tensorflow-GPU

先安装好anaconda:
在清华镜像下载想要的版本的安装包
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
解压后执行安装:
bash Anaconda2-xxx-Linux-x86_64.sh
yes
source ~/.bashrc

版本对应:
CUDA — cuDNN ---- — tf版本
8.0 ------- V6.0/V6.1 ------ --1.4 -->这个版本首选
9.0 ----- V7.1 --------------- 1.6/1.5

安装nvidia driver:
sudo lshw -numeric -C display #检查显卡对应的驱动型号 10DE:1C03

http://www.nvidia.com/Download/index.aspx

ubuntu-drivers devices #查看可以实用的驱动

sudo apt-get install nvidia-XXX
sudo apt-get install mesa-common-dev
sudo apt-get install freeglut3-dev

然后重启电脑查看显卡驱动是否安装好

安装CUDA8:
https://developer.nvidia.com/cuda-downloads

sudo sh cuda_XXX_linux.run --override   # ctrl+c accept no yes...

配置CUDA环境变量:
sudo gedit ~/.bashrc
#在bashrc的最后添加下面两行,添加完退出保存
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda-8.0

将环境生效:

source ~/.bashrc
nvidia-smi #查看显卡内容

安装CUDNN:
https://developer.nvidia.com/cudnn #帐号密码:1364245724@qq.com Wu483256
tar xvzf cudnn-xxx_.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda-8.0/lib64/libcudnn*

安装tensorflow:
去清华镜像下载对应的版本
https://mirrors.tuna.tsinghua.edu.cn/help/tensorflow/
或者

pip install tensorflow-gpu==1.4(版本号)

测试tensorflow:
$python

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

pip升级失败以及anaconda的pip/conda安装提示权限不够的错误的解决方案 :

是因为anaconda的安装目录的拥有者和用户组都是root,本地用户权限不够,只需让自己拥有相应的权限就好了

【解决方案】

输入一下命令,进入到anaconda安装目录的父目录

[html] view plain copy

sudo chmod 777 -R anaconda2

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值