【tensorflow】0.环境搭建

news/2025/1/16 7:53:36/

本文目标为搭建tensorflow-gpu环境。最终的环境为python 3.6 + cuda9.0 + cuDNN7.1 + tensorflow1.8-gpu

由于电脑之前安装了tensorflow的cpu版本,现在要使用gpu版本,需要卸载已安装的cpu版本。由于我的python环境是anaconda集成环境,之前tensorflow安装采用的是pip安装,这里卸载直接采用pip卸载

pip uninstall tensorflow

本机环境:

  • windows 10 , 64位系统
  • i5-7500HQ
  • NVIDIA GeForce GTX 1050
  • Python:Anaconda集成环境,Python 3.6.2

安装CUDA

tensorflow官网要求最新的tensorflow-gpu应该使用cuda9.0版本。

运行支持 GPU 的 TensorFlow 所需满足的要求

  • CUDA® 工具包 9.0。如需了解详情,请参阅 NVIDIA 的文档。请务必按照 NVIDIA 文档中的说明将相关的 CUDA 路径名附加到 %PATH% 环境变量中。
  • cuDNN v7.0。如需了解详情,请参阅 NVIDIA 文档。请注意,cuDNN 通常安装在与其他 CUDA DLL 不同的位置。请务必将您安装了 cuDNN DLL 的目录添加到 %PATH% 环境变量中。
  • CUDA 计算能力为 3.0 或更高的 GPU 卡
  • 要使用不同版本的 cuDNN,您必须从源代码构建。

在NVIDIA官网的CUDA版本列表找到9.0下载。传送门

安装的时候有个槽点:由于我需要的python的环境,所以不需要整合Visual Studio,安装CUDA的时候一定不要勾选 Visual Studio Integration(这里需要自定义安装,默认的精简安装会选择这个选项!)。貌似这个Visual Studio Integration会匹配你的VS(本机装了VS2017),如果VS版本不对就会安装失败。所以强烈建议,不使用VS整合Tensorflow的同学一定不要勾选Visual Studio Integration

安装完成后,会自动设置好环境变量。CUDA_PATHCUDA_PATH_V9_0。如下图
这里写图片描述

可以在命令行检查环境是否正确。在命令行下输入:nvcc --version,若有版本输出,则表明cuda环境是OK的,我的输出如下:

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Sep__1_21:08:32_Central_Daylight_Time_2017
Cuda compilation tools, release 9.0, V9.0.176


安装cuDNN

安装cuDNN需要注册登陆NVIDA DEVELOPER网站。由于我们的cuda版本是9.0,故cuDNN选择版本cuDNN v7.1.4 Library for Windows 10
这里写图片描述

下载得到一个压缩文件:cudnn-9.0-windows10-x64-v7.1_2.zip。

配置cuDNN环境变量,两种方案,一种NVIDIA cuDNN Guide给出,一种Tensorflow Installation Guide给出。

1、Tensorflow方案(我采用这种方式,亲测可用)

cuDNN 通常安装在与其他 CUDA DLL 不同的位置。请务必将您安装了 cuDNN DLL 的目录添加到 %PATH% 环境变量中

为了保险起见,我在系统PATH环境变量中,添加了3个关于cuDNN的目录。
这里写图片描述

2、NVIDIA方案(和CUDA环境交叉)

  • 解压缩
  • 复制 <installpath>\cuda\bin\cudnn64_7.dllC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin目录下
  • 复制 <installpath>\cuda\ include\cudnn.hC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include目录下
  • 复制 <installpath>\cuda\lib\x64\cudnn.libC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64目录下
  • 设置cuDNN环境变量:配置过CUDA_PATH即可。
  • cuDNN installation Guide

安装Tensorflow-gpu

tensorflow installation guide,有时打不开,需要翻墙

在 Anaconda 内部,我们建议使用 pip install 命令来安装 TensorFlow,而不要使用 conda install 命令。

要安装 GPU 版本的 TensorFlow,请输入以下命令:

C:\> pip3 install --upgrade tensorflow-gpu

验证安装:

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

输出:

T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:1356] Found device 0 with properties:
name: GeForce GTX 1050 major: 6 minor: 1 memoryClockRate(GHz): 1.493
pciBusID: 0000:01:00.0
totalMemory: 2.00GiB freeMemory: 1.61GiB

b’hello,tensorflow!’


http://www.ppmy.cn/news/788916.html

相关文章

汇编点亮LED I.MX6U-ALPHA

汇编点亮LED I.MX6U-ALPHA 硬件所使用的是正点原子的I.MX6U-ALPHA开发板 创建led.s文件&#xff0c;将以下代码写入&#xff1a; .global _start /* 全局标号 *//** 描述&#xff1a; _start函数&#xff0c;程序从此函数开始执行此函数完成时钟使能、* GPIO初始化、最终…

使用Mx-Yolov3与Maixhub,本地及线上训练K210模型文件

声明&#xff1a;本文谢绝转载&#xff01; CSDN地址&#xff1a;https://blog.csdn.net/weixin_58068682 个人博客&#xff1a;https://amnesia-f.gitee.io/ 一、功能介绍 目前提供两种训练&#xff1a; 目标分类&#xff1a; 识别图片所属的种类&#xff0c; 比如图中是苹…

因果推断Causal Inference: What If (the book)没有模型的因果推理部分章节结构

第一章 因果效应的定义 1.1 个体因果效应&#xff1a;介绍了个体因果效应的概念和定义&#xff0c;即在给定某个干预条件下&#xff0c;个体结果的变化量。1.2 平均因果效应&#xff1a;介绍了平均因果效应的概念和定义&#xff0c;即在给定某个干预条件下&#xff0c;总体结果…

i.MX8M系列开发板开发资料(米尔MYD-JX8MX)

1. i.MX8M开发板概述NXP 公司的 i.MX8M 系列的应用处理器基于 Arm Cortex-A53 和 Cortex-M4 内核, 具有业界领先的音频、语音和视频处理功能,适用于从消费家庭音频到工业楼宇自动化及移 动计算机等广泛应用。 作为 NXP 官方合作的设计公司,米尔电子推出了基 于 NXP 公司 i.M…

i.mx嵌入式linux编程,基于i-mx287 ARM9的嵌入式linux上 交叉编译paho.mqtt.embedded-c库

本文参考https://blog.csdn.net/sqshining/article/details/70193354 参考上述连接进行自己的ARM9开发板上实现paho.mqtt.embedded-c库的编译,实现测试例程在周立功的i-mx287A开发板上正常运行,在交叉编译mqtt库期间遇到一些坑,在此记录并说明一番,给自己的学习之路添加一些…

MX350显卡+win10安装TensorFlow(anaconda下安装)

环境配置&#xff1a; 系统cpugpuCUDACUDNNwin10i5-10210UMX350显存2GB10.2v7.6.5 一、安装CUDA 1.确认计算机显卡型号&#xff1a; 在设备管理器&#xff08;鼠标右键单击此电脑&#xff0c;选择管理&#xff0c;进入页面&#xff0c;即可在左侧看到设备管理器&#xff09…

mx-framework前端框架使用经验积累(SG-UAP)

在使用SG-UAPV2.8及mx-framework前台框架开发项目的过程中&#xff0c;积累了一些前端框架的使用技巧&#xff0c;基本上涵盖了大部分常用点。如有需要可以打开“阅读更多”后&#xff0c;直接全文搜索你想知道的问题。若未查询到可以留言告诉我&#xff0c;我也会尽力解答。 *…

对应的cuda版本 显卡驱动版本_NVIDIA显卡,显卡驱动版本,CUDA版本,cudnn版本之间兼容关系及如何选择...

要配置NVIDIA显卡的CUDA和cudnn&#xff0c;各种版本之间的依赖关系以及与其他使用GPU的库版本兼容一直没有弄明白&#xff0c;最近经过多次卸载重装&#xff0c;终于成功配置好了显卡计算环境&#xff0c;于是把各个驱动程序和库之间的依赖关系理一下。 一.显卡 我的显卡如下:…