简介

这篇博客Ubuntu16.04+CUDA9+tensorflow的安装流程。不过，随着软件不断更新，现在Ubuntu18.04逐渐成为客户端的主流，加上tf开始支持CUDA10，在这里在更新一下教程。

看一下tensorflow官方版本的支持：

确定安装1.13.1的版本

安装CUDA10.0

下载CUDA10：https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=deblocal

转到下载的目录执行：

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}}

安装cuDNN 7.4

在这里下载cuDNN7.4：https://developer.nvidia.com/rdp/cudnn-archive
需要登录下载

然后在下载目录下执行：

sudo dpkg -i libcudnn7_7.4.2.24-1+cuda10.0_amd64.deb

安装Tensorflow-GPU版本

这里默认是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) +by_ = 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.33feed = {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

看到一系列的输出，表示安装成功！！

Ubuntu18.04 + CUDA10.0 + tensorflow-gpu 安装过程相关推荐

NVIDA GTX2080 Ubuntu18.04 cuda10.0 cudnn7.5 tensorflow 1.12 install
NVIDA GTX2080 Ubuntu18.04 cuda10.0 cudnn7.5 tensorflow 1.12 install 心仪已久的显卡2080终于到了,Ubuntu18这么帅,怎么还能 ...
深度学习环境配置 (Ubuntu18.04 + CUDA10.0 + cuDNN7.6.5 + TensorFlow2.0)
@ Bergen, Norway 第一次安装 CUDA 的过程简直抓狂,中间出现了很多次莫名其妙的 bug,踩了很多坑.比如装好了 CUDA 重启后进不去桌面系统了,直接黑屏.比如鼠标键盘都不 wor ...
ubuntu16.04+GTX1060+ CUDA9.0+CUDANN7.0 tensorflow(GPU)安装，实测有效，安装包我放在云盘中，直接下载使用。
最近在运行一些 semantic segmentation 的代码,好多都是GPU版本的,自己就安装一下,记录一下,开始错了好多次,最后参考了一些文章,发现这个版本问题很重要,我用的是一下版本cuda ...
Ubuntu18.04+CUDA10.0+GTX1050Ti安装
安装Nvidia显卡驱动 sudo ubuntu-drivers autoinstall 安装完成后,使用命令nvidia-smi就可以查看到显卡的信息了,以960M为例: 安装CUDA10.0 1. ...
linux 显卡扩展坞,Ubuntu18.04上外接显卡扩展坞安装Nvidia驱动和CUDA10.0及cuDNN
前言一通折腾,算是把显卡扩展坞给接上用起来了.能找到的Ubuntu外接显卡扩展坞配置深度学习环境的资料比较少,乱折腾一番总算是能用了.现努力回忆过程,尽可能完整地记录下来配置环境的过程. ----- ...
Ubuntu18.04+cuda10.1+cudnn7.6.3+tensorrt5.1.5.0（deb）
环境配置Ubuntu18.04+cuda10.1+cudnn7.6.3+tensorrt5.1.5.0 Ubuntu18.04安装本系统配置背景 1. cuda10.1安装 2. 驱动安装 3. c ...
惠普暗夜精灵4双系统ubuntu18.04+CUDA10.1+cudnn+pytorch+tensorflow+ros
安装ubuntu18.04: 本机器型号安装ubuntu没有什么疑难杂症,跟着网上大佬的教程,下载镜像文件,做启动盘安装即可. 分区方案: efi 2G swap 16G / 500G 启动引导选择e ...
基于Windows环境下CPU和GPU版本Tensorflow详细安装过程
基于Windows环境下CPU和GPU版本Tensorflow详细安装过程一.Tensorflow简介 TensorFlow 是一个开源的.基于 Python 的机器学习框架,它由 Google 开 ...
caffe运行不停止_caffe(gpu)安装过程及问题解决
2019.12.05 caffe(gpu)安装参考网址: 教程1: weiliu89/caffegithub.com 教程2: https://blog.csdn.net/yggaoeecs/ar ...

Ubuntu18.04 + CUDA10.0 + tensorflow-gpu 安装过程

简介

安装CUDA10.0

安装cuDNN 7.4

安装Tensorflow-GPU版本

Ubuntu18.04 + CUDA10.0 + tensorflow-gpu 安装过程相关推荐

最新文章

热门文章