欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Windows 下配置Tensorflow

程序员文章站 2022-05-09 13:02:20
...

选择安装哪一种TensorFlow

首先根据自己的硬件配置选择一种

TensorFlow with CPU support only. 如果你的计算机没有 NVIDIA 的独立显卡, 你就只能选择安装CPU版本.
TensorFlow with GPU support. 计算机有NVIDIA的独立显卡的话, 就安装GPU版本吧, 这样训练速度将比CPU 快的多.

GPU 版本 TensorFlow 的配置需求

Anaconda

这里使用 Anaconda 来安装 TensorFlow,首先可以去官网下载. 不过如果觉的下载速度慢的话, 可以去清华的源 下载, 速度非常快.
安装好 Anaconda 后,就可以按照下面步骤安装 TensorFlow 了.

  1. 打开 Anaconda Prompt
  2. 输入 conda create -n tensorflow python=3.6 创建一个名为 tensorflow的环境
  3. **环境 activate tensorflow
  4. 使用命令 pip install --ignore-installed --upgrade tensorflow-gpu 进行安装.

国内的网络使用 conda 下载各种包是非常慢的. 下面说明如何修改到国内下载源.
输入命令

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

这时, 在你的用户名目录下, 有一个 .condarc 文件, 使用记事本打开.

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - defaults
show_channel_urls: true

删除掉第三行, 这时再使用 conda install 各种东西, 你会发现速度非常快.

下面是一些 conda 常用命令

# 创建一个名为 envname 的环境
conda create -name envname

# **名为 envname 的环境
activate envname

# 列出所有环境
conda info --envs

# 删除名为 envname 的环境
conda remove -n envname --all

关于 Anaconda 的更多使用问题 , 可以去读官方文档.

验证安装

为了验证tensorflow 安装是否正确,运行下面一段代码

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

如果输出了 Hello, TensorFlow, 安装就没问题了.