发布于2019-08-06 10:41 阅读(1348) 评论(0) 点赞(1) 收藏(4)
这篇博客Ubuntu16.04+CUDA9+tensorflow的安装流程。不过,随着软件不断更新,现在Ubuntu18.04逐渐成为客户端的主流,加上tf开始支持CUDA10,在这里在更新一下教程。
看一下tensorflow官方版本的支持:
确定安装1.13.1的版本
转到下载的目录执行:
sudo dpkg -i cuda-repo-ubuntu1804-10-0-local-10.0.130-410.48_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-10-0-local-10.0.130-410.48/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda
在当前终端设置环境变量:
export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
在这里下载cuDNN7.4:https://developer.nvidia.com/rdp/cudnn-archive
需要登录下载
然后在下载目录下执行:
sudo dpkg -i libcudnn7_7.4.2.24-1+cuda10.0_amd64.deb
这里默认是python3的环境。
转到home下,执行:
sudo apt-get install python3-pip python3-dev python-virtualenv
确保安装了基本的工具。
之后,执行创建虚拟环境的命令:
virtualenv --system-site-packages -p python3 ~/tensorflow
激活虚拟环境:
source ~/tensorflow/bin/activate
更新pip
easy_install -U pip
在虚拟环境下,执行安装:
pip3 install --upgrade tensorflow-gpu
之后需要确认安装NVIDIA驱动!!
sudo ubuntu-drivers autoinstall
测试安装,在当前终端下,vim test.py
,并添加内容:
import tensorflow as tf
import numpy as np
x = tf.placeholder("float",shape=[None,1])
W = tf.Variable(tf.zeros([1,1]))
b = tf.Variable(tf.zeros([1]))
y = tf.matmul(x,W) +b
y_ = tf.placeholder("float",[None,1])
cost = tf.reduce_sum(tf.pow((y_-y),2))
train_step = tf.train.GradientDescentOptimizer(0.001).minimize(cost)
init = tf.initialize_all_variables()
sess = tf.Session()
sess.run(init)
All_x = np.empty(shape=[1,1])
All_y = np.empty(shape=[1,1])
for i in range(1000):
x_s = np.random.rand(1,1)
y_s = np.dot([[0.33]],np.random.rand(1,1)) + 0.33
feed = {x: x_s, y_: y_s}
sess.run(train_step,feed_dict=feed)
print("After %d iteration:"%i)
print("W : %f"%sess.run(W))
print("b : %f"%sess.run(b))
All_x = np.concatenate((All_x,x_s))
All_y = np.concatenate((All_y,y_s))
print(All_x)
print(All_y)
之后,执行:
python3 test.py
看到一系列的输出,表示安装成功!!
作者:嘴巴嘟嘟
链接:https://www.pythonheidong.com/blog/article/7822/504493e0c3252dc5ba13/
来源:python黑洞网
任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任
昵称:
评论内容:(最多支持255个字符)
---无人问津也好,技不如人也罢,你都要试着安静下来,去做自己该做的事,而不是让内心的烦躁、焦虑,坏掉你本来就不多的热情和定力
Copyright © 2018-2021 python黑洞网 All Rights Reserved 版权所有,并保留所有权利。 京ICP备18063182号-1
投诉与举报,广告合作请联系vgs_info@163.com或QQ3083709327
免责声明:网站文章均由用户上传,仅供读者学习交流使用,禁止用做商业用途。若文章涉及色情,反动,侵权等违法信息,请向我们举报,一经核实我们会立即删除!