前言:
之前安装的 CPU 版本的 tensorflow 一直出问题,索性就直接安装 GPU 版本的 tensorflow 了(有了GPU 就不能浪费)。

安装过程:

1)看自己有无 GPU,找到对应 GPU 的版本:任务管理器 => 性能 => GPU

然后在 Anaconda Prompt 里面输入 nvidia-smi 来检查是否含有英伟达驱动,若没有则需要在 英伟达官网 安装驱动:


下载好了驱动之后,在 cmd 中输入命令 nvidia-smi


2)在虚拟环境中安装 GPU 加速软件包 cuDNN、CUDA:

首先根据官网 https://tensorflow.google.cn/install/source_windows?hl=en#gpu 来确定 tensorflow、cuDNN、CUDA 的对应关系:

然后根据网站 https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html 来确定自己的 GPU 所支持的 CUDA 版本:

这里我安装的是 tensorflow-gpu==2.1、cuDNN=7.6、CUDA=10.1

# 创建名为 tensorflow 的虚拟环境,其 python 版本为 3.7
conda create -n tensorflow python=3.7
# 进入 tensorflow 的虚拟环境
conda activate tensorflow
# 安装英伟达的SDK10.1版本
conda install cudatoolkit=10.1
# 安装英伟达深度学习软件包7.6版本
conda install cudnn=7.6
# 安装 tensorflow-gpu 指定 2.1 版本
pip install tensorflow-gpu==2.1

2.0以后tensorflow和tensorflow-gpu有啥区别:conda 装东西总是喜欢检查环境中所有包的依赖关系,当你装了 tensorflow 再用 conda 装matplotlib,就把 tensorflow 的 numpy 包给替换了,造成使用 tensorflow 后报 numpy 错误。因为安装软件包的话,能用 pip install xxx 装尽量用 pip,少用 conda install xxx。


其他额外补充:

1、tensorflow 设置 GPU 训练模型

import tensorflow as tfgpus = tf.config.list_physical_devices("GPU")
if gpus:tf.config.experimental.set_memory_growth(gpus[0],True)
// 测试 GPU 是否可用
import tensorflow as tf
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
version=tf.__version__  #输出tensorflow版本
gpu_ok=tf.test.is_gpu_available()  #输出gpu可否使用(True/False)
print("tf version:",version,"\nuse GPU:",gpu_ok)
tf.test.is_built_with_cuda()  # 判断CUDA是否可用(True/False)
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))

2、报错UnavailableInvalidChannel: HTTP 404 NOT FOUND for channel Esri https://api.anaconda.org/Esri

一般出现这种错误都是镜像源问题,需要更改镜像源。虽然我在最开始的时候就使用了清华的镜像源,可能在后面出现了错误,所以我们再更新一遍清华的镜像源即可。

参考:conda安装指定版本TensorFlow 中更换镜像源的方法。

此报错主要是由于安装高版本的cudatoolkit、cudnn导致的,再更换为清华的镜像源之后此错误会消失。


3、tensorflow-gpu 版本也要与 keras 版本相对应,不然也会产生报错

如我使用的是 tensorflow_gpu==2.6.0,那么我使用的 keras 版本也应该是 2.6.0,使用命令pip install keras=2.6.0,这样搭配起来不会报错。关于 tensorflow 与 keras 版本对应关系,目前还没到最新的,等找到了再更新到此。

安装 GPU 版本的 tensorflow 完整版本相关推荐

  1. 真实机下 ubuntu 18.04 安装GPU +CUDA+cuDNN 以及其版本选择(亲测非常实用)【转】...

    本文转载自:https://blog.csdn.net/u010801439/article/details/80483036 ubuntu 18.04 安装GPU +CUDA+cuDNN : 目前, ...

  2. Ubuntu 18.04从源代码编译安装GPU支持的Tensorflow 1.8.0

    当我第一次使用Linux时,我想要做的第一件事就是安装GPU版本Tensorflow.我找到了一个很好的指南.但即使考虑到这一点,我也花了超过40个小时的安装.这篇文章我想节省你的时间,分享自己的经验 ...

  3. Linux复习资料——CentOS7下安装MySQL5.7.22(完整版本)

    目录 下载地址: CSDN下载地址 xftp上传包 解压包 批量安装: 启动mysql,并查看运行状态,显示初始密码 登录MySQL 修改密码与刷新: 外网链接 下载地址: https://downl ...

  4. python 查看 nvida 驱动、 cuda、pytorch、tensorflow的版本

    nvidia 显卡.驱动.cuda 的版本 import subprocess command = 'nvidia-smi' # 终端命令 open_process = subprocess.Pope ...

  5. windows10 安装gpu版本TensorFlow脑壳疼自我记录

    写文章时间为:2018.10.26 显卡为NVIDIA GeForce RTX 2080Ti(刚出没有多久),系统为window10 放在开始想说的是,以下是我安装过程,仅是自我总结,仅供参考. 再之 ...

  6. tensorflow 安装GPU版本,CUDA与cuDNN版本对应关系,RTX3050Ti (notebook)

    前言 安装Tensorflow-gpu 与 keras的时候,一定先要注意版本的对应,不然很容易出错,在看的时候,建议先看完整篇文章再上手. 一.环境+配置 本机环境 显卡:RTX3050Ti(not ...

  7. tensorflow运行环境linux,在ubuntu或者min运行环境下安装gpu版本的tensorflow

    转载请注明出处:blog.csdn.net/sproll 本文描述在ubuntu系列操作系统上安装gpu版本的tensorflow的过程. 0,准备工作 BIOS中关闭板载显卡,显示器接在nvidia ...

  8. 【TensorFlow】——Windows10、pycharm、Python3.6.4中安装CPU和GPU两种版本的TensorFlow

    tensorflow分为两种,一种是cpu版本,一种是gpu版本 CPU版本可以直接通过pip进行安装: 这里是在国内镜像源中进行的下载,下载的版本是2.2.0版本 pip --default-tim ...

  9. 安装指定版本的GPU版本的tensorflow小技巧

    安装指定版本的GPU版本的tensorflow小技巧 使用类似 pip install tensorflow-gpu==1.4 命令即可

最新文章

  1. 用python操作mysql数据库(之“更新”操作)
  2. amcharts 网页绘图插件
  3. 令人惋惜的天才新秀:16岁上剑桥大学,27岁就出名,数学事业一路畅通无阻,但自从结婚后,人生从此翻天覆地······
  4. 匹配中文字符的正则表达式: [u4e00-u9fa5](
  5. FreeSql (三)实体特性
  6. 《汇编语言》王爽—实验五详解
  7. 【Level 09】U1 The way I see it L3 At your service
  8. 基于JAVA+SpringMVC+Mybatis+MYSQL的网上书城
  9. Unix原理与应用学习笔记----第四章 文件系统2
  10. python sizeof_python 变量作用域 v.__sizeof__() python 深复制 一切皆对象 尽量减少内存消耗 赋值语句的原理...
  11. Oracle 10g 中 X$KCVFH 说明
  12. Django 系列博客(二)
  13. 计算机如何删除用不到的打印机驱动程序,win7卸载不了打印机驱动怎么办
  14. sqlserver行列转换,动态行转换
  15. 【CSS】纯CSS实现三级导航(模板)
  16. WIN11右键菜单默认展开
  17. 源码分析:《Topic-to-Essay Generation with Neural Networks》
  18. 技嘉B550M-AORUS-ELITE主板RAID故障与解决方案
  19. 推荐算法实践-章节三-推荐系统冷启动问题-阅读总结
  20. BZOJ 1502 NOI2005 月下柠檬树 Simpson自适应公式

热门文章

  1. Python练习题——统计字符串中分别有多少个字母数字空格及特殊字符
  2. Springboot简单介绍
  3. 分布式架构之系统拆分
  4. ElGamal体制和Diffie-Hellman问题
  5. MSDN I tell you下载方式
  6. 论软件的系统测试以及应用-系统分析师-2022年上半年必考知识点
  7. 【OpenCV DNN】Flask 视频监控目标检测教程 02
  8. ECharts自定义toolbox中增加自定义按钮
  9. oracle select查询慢,从select结果集中查询太慢
  10. Watch Out!(中文名:贱鸟跳跳)