TensorFlow学习记录(一)—— 安装

文章目录

  • 前言
  • 一、Anaconda 3 下载安装
  • 二、确定一下你的GPU是否支持CUDA
    • 1.查看GPU
    • 2. 必应 / google搜索:NVIDIA XXXXXX + SPECIFICATION
  • 三、CUDA下载安装
    • 1.直接搜索CUDA 10.0进行下载
    • 2.安装好后,打开anaconda的terminal输入
  • 四、cuDNN下载安装
    • 1.下载cuDNN(需要先注册一下)
    • 2.解压得到:
    • 3. 将bin 、include、 lib\x64 下面的文件复制到 CUDA安装文件夹对应的bin 、include、 lib\x64 下面完成安装
    • 4.测试一下
  • 五、用ANACONDA安装TENSORFLOW
    • 1. 在ANACONDA里创建名为TENSORFLOW的环境(你可以叫他任何名字,这里我叫这个环境为TENSORFLOW
    • 2. 发出相关命令以在 conda 环境中安装 TensorFlow
  • 总结 + 安装包


前言

tensorflow学习中的一些记录…嘻嘻
(一) 下载与安装
Windows 10 + Anaconda 3 + Python 3.7 + CUDA toolkit 10.0 + cuDNN v7.4+ tensorflow 1.14


一、Anaconda 3 下载安装

下载地址:

https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/?C=M&O=D

不要挑最新的,下载下来安装界面会提示require python 3.8适配的python 3.8, 这里需要注意整体搭配,看要安装的软件是否支持python版本

tensorflow的官网上可以查到型号,这里我参考如下图:
TensorFlow学习记录(一)—— 安装_第1张图片

我使用安装的搭配为:
tensorflow 1.14、 python 3.7, CUDA 10.0,cuDNN 7.4. (这里为什么不搭配tensorflow 2.0,最后会解释)

那就找Anaconda 3支持python3.7的版本,我从下载地址里挑选版本为:
TensorFlow学习记录(一)—— 安装_第2张图片
下下来安装就完事,要注意 添加环境变量
TensorFlow学习记录(一)—— 安装_第3张图片
成功过后,测试一下,打开cmd命令行:
TensorFlow学习记录(一)—— 安装_第4张图片
或者在 cmd 中输入 conda --version 检测anaconda环境是否安装成功。


二、确定一下你的GPU是否支持CUDA

1.查看GPU

TensorFlow学习记录(一)—— 安装_第5张图片

2. 必应 / google搜索:NVIDIA XXXXXX + SPECIFICATION

即可查看版本是否支持
TensorFlow学习记录(一)—— 安装_第6张图片


三、CUDA下载安装

1.直接搜索CUDA 10.0进行下载

TensorFlow学习记录(一)—— 安装_第7张图片

2.安装好后,打开anaconda的terminal输入

(直接cmd打开输入也行)

nvcc -V

进行测试,显示如下证明安装成功
TensorFlow学习记录(一)—— 安装_第8张图片


四、cuDNN下载安装

1.下载cuDNN(需要先注册一下)

https://developer.nvidia.com/rdp/cudnn-archive

挑选 7.4 版本,并且适用于CUDA 10.0 这里要对应
TensorFlow学习记录(一)—— 安装_第9张图片

2.解压得到:

TensorFlow学习记录(一)—— 安装_第10张图片

3. 将bin 、include、 lib\x64 下面的文件复制到 CUDA安装文件夹对应的bin 、include、 lib\x64 下面完成安装

CUDA安装文件夹为:
TensorFlow学习记录(一)—— 安装_第11张图片

4.测试一下

打开cmd 进入文件夹:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\demo_suite

运行两个程序(deviceQuery / bandwidthTest)显示如下即可。
TensorFlow学习记录(一)—— 安装_第12张图片
TensorFlow学习记录(一)—— 安装_第13张图片

五、用ANACONDA安装TENSORFLOW

1. 在ANACONDA里创建名为TENSORFLOW的环境(你可以叫他任何名字,这里我叫这个环境为TENSORFLOW

打开软件:
TensorFlow学习记录(一)—— 安装_第14张图片
输入

conda create -n tensorflow pip python=3.7

注意 pip python=3.X 的意思是在名为tensorflow的环境里搭建版本是3.X的python,改为你对应的版本

2. 发出相关命令以在 conda 环境中安装 TensorFlow

(tensorflow) C:\Users\JackieZ>pip install --ignore-installed --upgrade tensorflow-gpu==1.14

这里不加“==1.14”就会默认安装了最新的 TensorFlow 2.5,安装2.0以上版本会存在与1.0的语句不兼容情况,如下:
TensorFlow学习记录(一)—— 安装_第15张图片
因此解决方法有两种

  • 最简单的:安装1.14,这也是比较常用的搭配,方便学习
  • 或者:
    只修改两条语句,就可以覆盖全部语句,把
   import tensorflow as tf

改为

import tensorflow.compat.v1 as tf

tf.disable_v2_behavior()

总结 + 安装包

后面开始学习咯 顶顶顶顶顶顶顶顶顶顶顶顶顶顶顶顶顶顶顶

附上tensorflow目前的版本,参考以下博客
https://blog.csdn.net/weixin_43760844/article/details/113477352

安装包链接附上~~~~(使用的 Anaconda3 + CUDA + cuDNN)

https://pan.baidu.com/s/1ChCsQ7jaZ12VJ7jpRkQWMA
提取码: wow6

你可能感兴趣的:(tensorflow)