使用Anaconda安装TensorFlow详细教程

一、Anaconda安装

可以参考笔者的这篇博客:Anaconda安装详细教程

二、准备工作

1、单击启动Anaconda Prompt创建新虚拟环境

在这里插入图片描述
2、在Anaconda Prompt依次执行以下命令conda create -n pytorch python=3.6,创建名字为tensorflow的虚拟环境,再通过命令conda info --envs可以查到已存在的虚拟环境(tensorflow存在其中)

在这里插入图片描述
3、再使用命令conda activate tensorflow激活刚刚建立好的虚拟环境tensorflow,可以看见小括号中由base环境切换到tensorflow环境

在这里插入图片描述

三、TensorFlow安装

1、TensorFlow版本选择

TensorFlow版本分为CPU版本和GPU版本,CPU版本一般电脑都可以安装,但是GPU版本只有电脑的显卡支持cuda才可以安装,这两个版本的区别主要在于运行速度,GPU版本的TensorFlow运行速度更快,所以若电脑显卡支持cuda,推荐安装GPU版本。本文以安装GPU版本的TensorFlow进行说明。

2、TensorFlow的GPU版本对应Python版本

所有博客都建议去Tensorflow文档查看,其网址为https://tensorflow.google.cn/install/source_windows,但是笔者确没有找到。根据笔者安装的经验,TensorFlow 1.x的GPU版本对应的Python版本在3.7以下,TensorFlow 2.0和TensorFlow 2.1的GPU版本对应的Python版本在3.7以下,TensorFlow 2.2的GPU版本对应的Python版本在3.8以上。

笔者建议在创建虚拟环境时即指定Python版本为3.6版本既可向下兼容TensorFlow 1.x的GPU版本,又可以兼容TensorFlow 2.x的GPU版本。此处指出笔者安装TensorFlow 的GPU版本时出现的一个问题,因为笔者的base环境的Python版本为3.8版本,在创建虚拟环境时默认为Python版本为3.8。随后直接安装了最新的TensorFlow 2.3 的GPU版本,在进行测试时输入语句

import tensorflow as tf

直接报错:module 'numpy' has no attribute 'object',具体原因不可知。

3、TensorFlow的GPU版本安装

(1)在刚才最小化的Anaconda prompt窗口中输入命令conda install -c aaronzs tensorflow-gpu==1.15.0(注意这是TensorFlow 1.x的最后一个版本)

(2)待下载完毕后,输入python,会显示python版本为3.6。之后进行TensorFlow的测试,输入以下代码:

import tensorflow as tf
import os
a = tf.constant(1.)
b = tf.constant(2.)
print(a+b)
print(tf.__version__)
print('GPU:', tf.config.list_physical_devices('GPU'))
print(tf.test.is_gpu_available())

输出结果为:

Tensor("add:0", shape=(), dtype=float32)
1.15.0
# 笔者在测试第三个print时有问题,大家可以自行测试
AttributeError: module 'tensorflow._api.v1.config' has no attribute 'list_physical_devices'
True

在这里插入图片描述
无需安装cuda和cuDNN!!


参考博客

(1)TensorFlow-GPU极简———不用安装cuda、cuDNN!!!
(2)使用anaconda安装tensorflow (windows10环境)
(3)Anaconda安装使用及tensorflow配置

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
社会演员多的头像社会演员多普通用户
上一篇 2023年3月5日 上午9:22
下一篇 2023年3月5日 上午9:23

相关推荐