前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
社区首页 >专栏 >安装 tensorflow 1.1.0;以及安装其他相似版本tensorflow遇到的问题;tensorflow 1.13.2 cuda-10环境变量配置问题;Tensorflow 指定训练时如何指定

安装 tensorflow 1.1.0;以及安装其他相似版本tensorflow遇到的问题;tensorflow 1.13.2 cuda-10环境变量配置问题;Tensorflow 指定训练时如何指定

作者头像
西湖醋鱼
发布于 2020-12-30 03:26:40
发布于 2020-12-30 03:26:40
72100
代码可运行
举报
运行总次数:0
代码可运行
代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
# 安装 2.7 环境
conda create -n python2.7 python=2.7.17
conda activate python2.7
代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
# 安装 1.1.0 gpu版本
pip  install tensorflow-gpu==1.1.0
代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
# 配置环境变量
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64 
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/extras/CUPTI/lib64 
代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
# tensorflow1.13.2 cuda-10 lib库配置; 因为tensorflow 1.13版本以上要求cuda 10
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:/usr/local/cuda/lib64:/home/v-yaxu/nccl/build/lib 

保持更新,更多内容请关注 cnblogs.com/xuyaowen; 

常用操作:

tensorboard --logdir=/tmp/tensorflow/mnist/logs/mnist_with_summaries/ 

参考链接:

https://www.cnblogs.com/chay/p/10472993.html

https://blog.csdn.net/weixin_35653315/article/details/71308137

"ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory."?

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64

(解决使用 tensorflow 使用过程中,libcublas 库,找不到的错误)

Tensorflow 指定训练时使用的GPU

场景:有一台服务器,服务器上有多块儿GPU可以供使用,但此时只希望使用第2块和第4块GPU,但是我们希望代码能看到的仍然是有两块GPU,分别编号为0,1,这个时候我们可以使用环境变量CUDA_VISIBLE_DEVICES来解决;

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" # cuda 模式按照进行gpu的性能进行排序,设置此环境变量,GPU的顺序,将按照 pci_bus_id编号来进行设置;gpu顺序;这样在cuda_visble_devices环境变量就可以按照pci编号来进行选择gpu了;
os.environ["CUDA_VISIBLE_DEVICES"] = "0" # 设置程序环境变量;

指定GPU占用显存:

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7) # 确保每个GPU使用的显存,不超过 0.7
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options)) 

Tesorflow 默认会占用gpu 所有缓存,设置按需占用GPU缓存方法:

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config)

GPU 设置参考链接:

https://homepages.uc.edu/~schreihf/uchenry/post/using-gpus/ (gpu设置)

https://www.jianshu.com/p/0816c3a5fa5c

https://www.cnblogs.com/darkknightzh/p/6591923.html

https://www.tensorflow.org/guide/gpu (tensorflow 官方关于gpu使用的说明文档)

https://github.com/tensorflow/docs/blob/r1.13/site/en/guide/using_gpu.md (官方关于 tf.1.13的gpu使用文档)

设置参考自网络,如果失效,请纠正;

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2019-10-31 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
暂无评论
推荐阅读
编辑精选文章
换一批
安装GPU加速的tensorflow 卸载tensorflow
Ubuntu 16.04 + cuda9.0 + cudnn7.0 或 Ubuntu 16.04 + cuda8.0 + cudnn5.1
全栈程序员站长
2022/06/25
1K0
安装GPU加速的tensorflow 卸载tensorflow
linux安装多个版本的cuda和cudnn,无sudo权限安装,pip install tensorflowgpu==1.5
无sudo权限,参考https://blog.csdn.net/weixin_41278720/article/details/81255265 CUDA Toolkit 9.0和cudnn 7
水球喵子
2022/03/24
1.4K0
linux安装多个版本的cuda和cudnn,无sudo权限安装,pip install tensorflowgpu==1.5
GTX 1080+Ubuntu16.04+CUDA8.0+cuDNN5.0+TensorFlow
GTX 1080+Ubuntu16.04+CUDA8.0+cuDNN5.0+TensorFlow 安装指导
CreateAMind
2018/07/25
8110
GTX 1080+Ubuntu16.04+CUDA8.0+cuDNN5.0+TensorFlow
配置tensorflow GPU 版本填坑路
MelonTeam
2018/01/04
1.5K0
配置tensorflow GPU 版本填坑路
双显卡笔记本安装CUDA+theano、tensorflow环境
原文出处:http://www.cnblogs.com/jacklu/p/6377820.html
用户7043923
2020/03/12
8240
Ubuntu 16.04下为TITAN 1080 显卡安装驱动及Gpu版TensorFlow|深度学习
近来入坑了TITAN 1080显卡,在Ubuntu 16.04下为装好驱动以使用Gpu版TensorFlow可不简单,踩了许多坑之后写下此篇为记录。 下载Cuda 按装官方教程,我们可以应该安装Cu
陆勤_数据人网
2018/02/28
1.4K0
Ubuntu 16.04下为TITAN 1080 显卡安装驱动及Gpu版TensorFlow|深度学习
TensorFlow开发环境搭建(Ubuntu16.04+GPU+TensorFlow源码编译)
作者:刘才权 编辑:田 旭 安装平台 1 平台 目前TensorFlow已支持Mac、Ubuntu和Windows三个主流平台(64位平台), 2 GPU vs CPU 在安装时可以选择安装版本是否
机器学习算法工程师
2018/03/06
1.8K0
TensorFlow开发环境搭建(Ubuntu16.04+GPU+TensorFlow源码编译)
安装tensorflow GPU版本–tensorflow-gpu版本与CUDA版本对应关系(持续更新,目前到TF2.7)「建议收藏」
不同版本的tensorflow-gpu与CUDA对应关系如下表所示(图片有点旧了,python版本是2.7和3.3-3.8):
全栈程序员站长
2022/08/28
12.2K0
安装tensorflow GPU版本–tensorflow-gpu版本与CUDA版本对应关系(持续更新,目前到TF2.7)「建议收藏」
Ubuntu 18.04上安装cuda「建议收藏」
$ lspci | grep -i nvidia 我的显示为Tesla P800
全栈程序员站长
2022/09/28
1.9K0
Ubuntu 18.04上安装cuda「建议收藏」
TASK 1 了解TensorFlow
首先,我的服务器分配的IP是10.0.3.153,端口是9380至9389,服务器的域名为ras.sysu.edu.cn
平凡的学生族
2019/05/25
8620
Linux下Caffe、Docker、Tensorflow、PyTorch环境搭建(CentOS 7)
本文介绍了如何在CentOS 7上安装NVIDIA CUDA工具包以及CUDNN,并使用Docker和Nvidia-Docker来安装深度学习环境。通过这些环境,可以方便地运行深度学习模型的训练、测试和部署。
Tyan
2017/12/28
3.2K1
Linux下Caffe、Docker、Tensorflow、PyTorch环境搭建(CentOS 7)
Ubuntu17.04安装TensorFlow1.2的GPU版本
本文介绍了在Ubuntu 17.04系统上安装TensorFlow 1.2的GPU版本的过程,包括安装NVIDIA的GPU-CUDA, cuDNN, libcupti-dev, 以及通过pip或spip安装TensorFlow-GPU版本。
王小雷
2018/01/02
1.4K0
Ubuntu17.04安装TensorFlow1.2的GPU版本
Linux & Windows TensorFlow 1.14 升级 2.2
准备尝试升级TensorFlow 1.14 到2.2,需要同时升级本地和服务器的环境,本文记录主要过程。 环境需求 当前TensorFlow最高版本 2.2.+ ,需要CUDA 10.1,cudnn 7.6 官网下载 :https://developer.nvidia.com/ 显卡驱动需要满足CUDA版本要求 CUDA与显卡驱动:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html TensorFlow-GPU
为为为什么
2022/08/05
5330
Ubuntu 安装 tensorflow-gpu 1.4 +CUDA 8.0 +cuDNN详细教程
作者 | fendouai 编辑 | 磐石 出品 | 磐创AI技术团队 【磐创AI导读】:本文详细介绍了tensorflow-gpu在Ubuntu下的安装步骤。欢迎大家点击上方蓝字关注我们的公众号:磐创AI。 硬件环境:NVIDIA GTX 980 Ti 系统环境:Ubuntu 16.04 64位 一.安装 NVIDIA驱动 1. 关闭 Secure Boot 具体如何禁用 BIOS 中的 Secure Boot 要根据主板的情况。 以华硕主板的禁用方法为例: 首先进入 BIOS,然后选择 Boot ,
磐创AI
2018/07/03
1.5K0
【TensorFlow实战——笔记】第3章:TensorFlow第一步_TensorFlow的编译及安装
不管哪种情况,我们都推荐使用Anaconda作为Python的环境,因为可以避免大量的兼容性问题。
石瞳禅
2018/09/18
7940
【TensorFlow实战——笔记】第3章:TensorFlow第一步_TensorFlow的编译及安装
使用Anaconda搭建TensorFlow-GPU环境
前言:      对于深度学习来说,各种框架torch,caffe,keras,mxnet,tensorflow,pandapanda环境要求各一,如果我们在一台服务器上部署了较多的这样的框架,那么各种莫名的冲突 会一直伴随着你,吃过很多次亏之后,慢慢的接触了Anaconda,真的是很爽的一个功能,来管理环境配置。我们进行tensorflow安装的时候,还是使用Anaconda,鉴于国内墙太高 ,我们使用了Tsinghua的镜像文件,清华大学的Anaconda介绍地址见:https://mirror.tun
Gxjun
2018/03/27
4.6K0
Ubuntu16.04下Nvidia+Cuda8.0+Dynet安装教程
Ubuntu16.04下Nvidia+Cuda8.0+Dynet安装教程 - WeiYang Bloggodweiyang.com
godweiyang
2020/03/24
6480
Ubuntu16.04下Nvidia+Cuda8.0+Dynet安装教程
Ubuntu18.04LTS下cuda10.0+cudnn7.5+TensorFlow1.13环境搭建
前言 之前写过cuda环境的搭建文章, 这次干脆补全整个深度学习环境的搭建. ---- 开发环境一览 CPU: Intel core i7 4700MQ GPU: NVIDIA GT 750M
sean_yang
2019/03/15
1.9K0
Ubuntu18.04LTS下cuda10.0+cudnn7.5+TensorFlow1.13环境搭建
一文上手Tensorflow2.0(四)
【磐创AI导读】:本系列文章介绍了与tensorflow的相关知识,包括其介绍、安装及使用等。本篇文章是本系列文章的最后一篇。查看上篇:一文上手Tensorflow2.0之tf.keras|三。在文末作者给出了答疑群的二维码,有疑问的读者可以进群提问。想要获取更多的机器学习、深度学习资源,欢迎大家点击上方蓝字关注我们的公众号:磐创AI。
磐创AI
2019/08/23
1.6K0
配置深度学习主机与环境(TensorFlow+1080Ti) | 第三章 CUDA与CUDNN安装
01 概念介绍 CUDA(Compute Unified Device Architecture 统一计算设备架构) CUDA(Compute Unified Device Architecture),是英伟达公司推出的一种基于新的并行编程模型和指令集架构的通用计算架构,它能利用英伟达GPU的并行计算引擎,比CPU更高效的解决许多复杂计算任务。 使用CUDA的好处就是透明。根据摩尔定律GPU的晶体管数量不断增多,硬件结构必然是不断的在发展变化,没有必要每次都为不同的硬件结构重新编码,而CUDA就是提供了一
用户1332428
2018/03/08
1.7K0
配置深度学习主机与环境(TensorFlow+1080Ti) | 第三章 CUDA与CUDNN安装
推荐阅读
相关推荐
安装GPU加速的tensorflow 卸载tensorflow
更多 >
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档
查看详情【社区公告】 技术创作特训营有奖征文