当前位置: 首页 > news >正文

安装 GPU 版本的 tensorflow 完整版本

前言:
之前安装的 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。

http://www.lryc.cn/news/8222.html

相关文章:

  • BOM编程-设置地址栏上的URL
  • 设计模式之原型模式与建造者模式详解和应用
  • C语言(函数和递归)
  • 快乐的shell命令行
  • 大数据面试题flume篇
  • 零信任-深信服零信任aTrust介绍(5)
  • UVa 1343 The Rotation Game 旋转游戏 IDA* BFS 路径还原
  • 硬件学习 软件Cadence day02 画原理图的基本操作 (键盘快捷键 , 原理图设计流程 , 从开始到导出网表流程)
  • 【python】基于Socket的聊天室Python开发
  • 2023想转行软件测试的看过来,你想要了解的薪资、前景、岗位方向、学习路线都讲明白了
  • TortoiseSVN的使用
  • 操作系统(day09) -- 连续分配管理方式
  • APISpace 带你一起走进西湖美景
  • 傻白探索Chiplet,Design Space Exploration for Chiplet-Assembly-Based Processors(十三)
  • 系统分析师真题2020试卷相关概念一
  • 20230215_数据库过程_渠道业务计算过程
  • 【C++】Expression的学习笔记
  • [数据库迁移]-MySQL常见问题
  • C语言编译过程
  • 前端学习 ---常用标签
  • 2023年PMP考试难不难?
  • Netty 入门
  • 收藏|一文掌握数据分析在企业的实际流程
  • 100ask_imx6ull 输出PWM
  • yolov5编译安卓APP:解决图像上全是检测框
  • 为什么我们需要地图?
  • 攻防世界1.新手练习区
  • Python进阶篇(二)-- Django 深入模型
  • ABAP SALV实现弹出ALV选择
  • git check-pick,git patch 与 git stash 详解