Tensorflow-gpu的方式

主要参考https://www.jianshu.com/p/9911dee2f00b

主要讲一下在Linux下安装tensorflow gpu版的两种方式：使用conda和pip安装。安装cpu版本比较简单，安装命令相同，但是省去了安装cuda和cudnn。

注意： 在安装tensorflow gpu版本前，首先要清楚不同版本tensorflow对应不同的cuda，cudnn版本，而不同的cuda版本对显卡驱动版本又有要求。这几个版本不符合要求是无法使用gpu版本的tensorflow，下面给出了官网说明。
tensorflow-gpu各版本与cuda，cudnn匹配查询:
https://tensorflow.google.cn/install/source?hl=en#linux
CUDA版本与显卡驱动版本匹配查询：
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

关于CUDA和cuDNN安装本文不再详说，CSDN上一搜一大堆。先在nvidia官网下载CUDA安装文件和cuDNN安装文件（需要登陆后才能下载）。这里给出安装教程链接：CUDA安装教程，cuDNN安装教程。

其中，特别注意cuDNN下载需要登录注册，由于是国外的网站，多种浏览器的下载速度均较慢。通过尝试，使用迅雷可以获得不错的下载速度（具体浏览器通过迅雷下载的设置自行百度）

1、使用conda安装tensorflow（推荐）:
先用conda查询可以安装的tensorflow版本，根据当前的python环境，查询到的版本会有不同，在tensorflow官网安装页面上可以查到不同tensorflow版本支持的python ，若要安装cpu版本的tensorflow直接将后面的-gpu去掉即可。

conda search tensorflow-gpu

安装所需的tensorflow版本，等号后面指定版本，若不指定默认安装最新版本。在安装tensorflow 2.0以上版本时会自动安装上依赖的cuda和cudnn，省去了另外安装cuda和cudnn，1.x的版本未做测试，1.x的版本能否自动搜索安装所需的cuda和cudnn待确定。

conda install tensorflow-gpu=1.15.0

2、使用pip安装tensorflow
pip安装就一行命令，等号后面可以指定版本，若不指定默认安装最新版本：

pip install tensorflow-gpu==1.15.0

安装好后检查tensorflow gpu版本是否可用，返回结果为True则可用。

import tensorflow as tf
print(tf.test.is_gpu_available())

其次可以参考
https://blog.csdn.net/hillyfighting/article/details/106633362

Tensorflow-gpu的方式相关推荐

Windows Tensorflow GPU安装
GPU资源对神经网络模型的训练很重要,应充分利用电脑的显卡资源,加快模型的训练速度.这里是本人安装tensorflow-gpu的过程,记录了安装的步骤以及在每个步骤中参考的资料以及所遇到的坑. 大体步 ...
Linux16.04配置tensorflow(GPU源码编译)并深入了解tensorboard
Tensorflow – Google推出的一个强大的"深度学习框架".于2015年11月在GIthub上开源,在2016年4月补充了分布式版本,并于2017年1月发布了1.0版本 ...
windows10下安装tensorflow(gpu版本)
windows10安装tensorflow的gpu版本(pip3安装方式) 前言: TensorFlow 有cpu和 gpu两个版本:gpu版本需要英伟达CUDA 和 cuDNN 的支持,cpu版本不 ...
Windows 10环境下TensorFlow(gpu版本)配置教程——[图解] [详细版][零基础]
Tensorflow环境下的深度学习框架的配置主要包含以下几步: 0.前言 1.PyCharm的安装步骤: 2.Python的安装步骤: 3.AnaConda的安装步骤: 4.CUDA的安装步骤: 5 ...
Windows 下无痛安装 TensorFlow (GPU版)
Windows 下无痛安装 TensorFlow 2 (GPU版) 一. 原材料硬件 CUDA算力>=3.5的NVIDIA显卡如果想看你的显卡是否符合要求, 移步至 https://deve ...
教你从头到尾利用DL学梵高作画：GTX 1070 cuda 8.0 tensorflow gpu版
教你从头到尾利用DL学梵高作画作者:七月在线开发/市场团队中的三人,骁哲.李伟.July.本教程主要由骁哲撰写,李伟负责Linux命令部分. 时间:二零一六年九月二十五日配置:GTX 1070 c ...
教你从头到尾利用DL学梵高作画 GTX 1070 cuda 8 0 tensorflow gpu版
分享一下我老师大神的人工智能教程!零基础,通俗易懂!http://blog.csdn.net/jiangjunshow 也欢迎大家转载本篇文章.分享知识,造福人民,实现我们中华民族伟大复兴! 教你从头 ...
python 虚拟环境 tensorflow GPU
拿到一个新的容器之后,怎么创建一个独立的GPU训练环境呢?之前弄的时候总是零零散散的,现在把它总结在这里,供自己以及有需要的朋友查阅. conda创建 1.1 下载anaconda wget -c h ...
tensorflow GPU 内存不够
tensorflow GPU 内存不够 from tensorflow as tf gpus = tf.config.list_physical_devices('GPU') if gpus:try: ...
TensorFlow数据读取方式：Dataset API，以及如何查看dataset:DatasetV1Adapter的方法
TensorFlow数据读取方式:Dataset API Datasets:一种为TensorFlow 模型创建输入管道的新方式.把数组.元组.张量等转换成DatasetV1Adapter格式 Dat ...

Tensorflow-gpu的方式

Tensorflow-gpu的方式相关推荐

最新文章

热门文章