ubuntu16.04 安装tensorflow(GPU、CPU)

链接:

https://mirrors.tuna.tsinghua.edu.cn/help/tensorflow/

【注:需要更改成清华源】

GPU安装

适用GTX1050,GTX1060,GTX1070,GTX1080等高级显卡系列

  1. 安装好系统后,更新驱动。系统设置-》软件和更新-》附加驱动-》更新nvidia和intel
  2. 安装cuda9

下载 cuda9.run,(deb格式容易出错误)。执行如下命令:

sudo sh cuda_9.0.176_384.81_linux.run


accept

n(不要安装driver)

y

y

y

安装完成后,设置环境变量。

设置环境变量:

ctrl+H 打开 (主目录/home/*/) .bashrc文件,添加

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

export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH  

##############注意 解决import tensorflow 错误的问题############## 

更改为:(这里运行,不会再出现错误。很抱歉。)

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

终端运行:source ~/.bashrc

检查:nvcc --version


3. 安装cudnn7

下载 for linux版本的,执行如下命令:

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

sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

4. 安装anaconda(python3.5)

bash Anaconda3-4.2.0-Linux-x86_64.sh


yes(设置环境变量)

5. 安装tensorflow-gpu(任意版本)

复制开始的蓝色链接上的命令。


CPU安装:

直接复制开始链接上的命令安装tensorflow-cpu。









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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值