Ubuntu18.04安装Tensorflow-gpu
写篇blog记录一下配置tensorflow-gpu开发环境
环境版本
系统:Ubuntu 18.04 LTS
GPU:GeForce GTX 1050 Mobile
Cuda: 9.0
Cudnn: 7.0.5
Tensorflow-gpu:1.8.0
Python:python3.6.5
安装Cuda Toolkit 9.0
另外,下面的两个Patch 我都有下载,都是用dpkg装的
去到你下载好的三个deb的文件夹中,在终端执行下面的语句:
sudo apt-key add /var/cuda-repo-9-0-local-cublas-performance-update-2/7fa2af80.pub sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb sudo dpkg -i cuda-repo-ubuntu1604-9-0-local-cublas-performance-update_1.0-1_amd64.deb sudo dpkg -i cuda-repo-ubuntu1604-9-0-local-cublas-performance-update-2_1.0-1_amd64.deb sudo apt-get update sudo apt-get install cuda-9.0
安装完成后,声明一下环境变量
sudo vim ~/.bashrc
在文件的最后加上环境变量
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
profile里也加上声明
sudo vim /etc/profile
重新执行刚刚修改的初始化文档
source ~/.bashrc source /etc/profile
验证一下CUDA版本:9.0
nvcc --version
安装cudnn 7.0.5
cudnn 7.0.5
在官网下载 cuDNN v7.0.5 (Dec 5, 2017), for CUDA 9.0 下载三个文件ubuntu16.04文件 runtime library. developer library, and code samples and user guide
sudo dpkg -i libcudnn7_7.0.5.11-1+cuda9.0_amd64.deb sudo dpkg -i libcudnn7-dev_7.0.5.11-1+cuda9.0_amd64.deb sudo dpkg -i libcudnn7-doc_7.0.5.11-1+cuda9.0_amd64.deb
在检验cuDNN之前,先安装freeimage library作为ministCUDNN示例代码的依赖关系。
sudo apt-get install libfreeimage3 libfreeimage-dev
开始检验cuDNN啦!
cp -r /usr/src/cudnn_samples_v7/ $HOME cd $HOME/cudnn_samples_v7/mnistCUDNN sudo make clean && sudo make ./mnistCUDNN
安装成功的话,终端会如下显示:
安装Tensorflow-gpu
终于开始进入主题了!
Python 3.n用pip3安装
sudo pip3 install tensorflow-gpu==1.8.0
run个小sample,测试一下
#Python 3 import tensorflow as tf hello = tf.constant('Hello, TensorFlow!') sess = tf.Session() print(sess.run(hello))
和官方给的运行结果不一样,输出的是byte string。
以上,
完结撒花!
低调大师中文资讯倾力打造互联网数据资讯、行业资源、电子商务、移动互联网、网络营销平台。
持续更新报道IT业界、互联网、市场资讯、驱动更新,是最及时权威的产业资讯及硬件资讯报道平台。
转载内容版权归作者及来源网站所有,本站原创内容转载请注明来源。
- 上一篇
day19_java基础加强_动态代理+注解+类加载器
一、动态代理 1.1、代理模式 什么是代理模式及其作用? Proxy Pattern(即:代理模式),23种常用的面向对象软件的设计模式之一。 代理模式的定义:为其他对象提供一种代理以控制对这个对象的访问。 在某些情况下,一个对象不适合或者不能直接引用另一个对象,而代理对象可以在客户端和目标对象之间起到中介的作用。 优点: (1) 职责清晰,真实的角色就是实现实际的业务逻辑,不用关心其他非本职责的事务,通过后期的代理完成一件完成事务,附带的结果就是编程简洁清晰。 (2) 代理对象可以在客户端和目标对象之间起到中介的作用,这样起到了保护目标对象的作用。 (3) 高扩展性。 结构: 一个是你要访问的真正的对象(目标类),另一个是代理对象,真正对象(目标对象)与代理对象实现同一个接口,先访问代理类再访问真正要访问的对象。 其实装饰(包装)设计模式就是静态代理。 静态代理模式简单案例: KindWomen.java(接口) package com...
- 下一篇
Protostuff 的Java序列化和反序列化
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_34173549/article/details/80589509 <dependency> <groupId>io.protostuff</groupId> <artifactId>protostuff-core</artifactId> <version>1.4.0</version> </dependency> <dependency> <groupId>io.protostuff</groupId> <artifactId>protostuff-runtime</artifactId> <version>1.4.0</version> </dependency> import java.util.Map; import io.protostuff.LinkedBuffer; import...
相关文章
文章评论
共有0条评论来说两句吧...
文章二维码
点击排行
-
Docker使用Oracle官方镜像安装(12C,18C,19C)
- Springboot2将连接池hikari替换为druid,体验最强大的数据库连接池
- CentOS8编译安装MySQL8.0.19
- Docker快速安装Oracle11G,搭建oracle11g学习环境
- SpringBoot2配置默认Tomcat设置,开启更多高级功能
- MySQL8.0.19开启GTID主从同步CentOS8
- CentOS7,8上快速安装Gitea,搭建Git服务器
- Jdk安装(Linux,MacOS,Windows),包含三大操作系统的最全安装
- SpringBoot2编写第一个Controller,响应你的http请求并返回结果
推荐阅读
最新文章
- CentOS6,CentOS7官方镜像安装Oracle11G
- Windows10,CentOS7,CentOS8安装Nodejs环境
- CentOS8编译安装MySQL8.0.19
- SpringBoot2整合Thymeleaf,官方推荐html解决方案
- 设置Eclipse缩进为4个空格,增强代码规范
- CentOS7,8上快速安装Gitea,搭建Git服务器
- Windows10,CentOS7,CentOS8安装MongoDB4.0.16
- CentOS7安装Docker,走上虚拟化容器引擎之路
- CentOS6,7,8上安装Nginx,支持https2.0的开启
- CentOS7编译安装Cmake3.16.3,解决mysql等软件编译问题