tensorflow安装过程

最新推荐文章于 2024-07-09 17:03:37 发布

原创最新推荐文章于 2024-07-09 17:03:37 发布 · 2.2k 阅读

CC 4.0 BY-SA版权

1 篇文章

订阅专栏

本文详细介绍如何通过Anaconda安装TensorFlow的过程，并演示了一个简单的TensorFlow程序，验证安装是否成功。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

进入官网，拉到最下面，根据你系统是64还是32位下载安装，一般win10都是64位。安装就按默认选项就行。

打开Anaconda prompt，然后

conda create -n tensorflow python=3.7

conda activate tensorflow

以下命令默认安装符合你的驱动的最新版tensorflow：
conda install tensorflow-gpu

如果想安装指定版本，比如1.10.0，使用以下命令：
conda install tensorflow-gpu=1.10.0

如果想安装CPU版本，使用以下命令：
conda install tensorflow

安装完毕后，我们来编写一个简单的程序来验证安装。

在命令行下输入 conda activate tf2 进入之前建立的安装有 TensorFlow 的 Conda 虚拟环境，再输入 python 进入 Python 环境，逐行输入以下代码：

import tensorflow as tf

A = tf.constant([[1, 2], [3, 4]])
B = tf.constant([[5, 6], [7, 8]])
C = tf.matmul(A, B)

print(C)

如果能够最终输出:

tf.Tensor(
[[19 22]
[43 50]], shape=(2, 2), dtype=int32)

说明 TensorFlow 已安装成功。运行途中可能会输出一些 TensorFlow 的提示信息，属于正常现象。