【技术专题】TensorFlow2 Python深度学习 - TensorFlow2环境安装

29 阅读1分钟

大家好,我是锋哥。最近连载更新《TensorFlow2 Python深度学习》技术专题。

QQ截图20260306194834.jpg 本课程主要讲解基于TensorFlow2的Python深度学习知识,包括深度学习概述,TensorFlow2框架入门知识,以及卷积神经网络(CNN),循环神经网络(RNN),生成对抗网络(GAN),模型保存与加载等。同时也配套视频教程 《2026版 TensorFlow2 Python深度学习 视频教程》

TensorFlow2官网:tensorflow.google.cn/?hl=zh-cn

我们首先新建一个纯净python项目,Python版本3.11,我们选择使用虚拟环境,防止第三方库版本冲突。

image.png

然后我们安装tensorflow库,最新版本2.20。这里有两种安装方式,第一种是 pip install tensorflow同时支持CPU和GPU。还有一种学习版本安装 pip install tensorflow-cpu仅支持CPU。

锋哥建议,我们一般学习的话,用第二种方式即可。反正写的代码都一样。商用部署的话,会用到专业的服务器,操作系统乌班图,CPU普通的即可。GPU是英伟达几万的显卡的那种,如RTX系列,如何训练规模巨大,直接上几十万的英伟达H100,A100这种,还可以搞多个集群。

CPU和GPU都支持

pip install tensorflow -i http://mirrors.aliyun.com/pypi/simple/   --trusted-host mirrors.aliyun.com

仅需要CPU支持

pip install tensorflow-cpu -i http://mirrors.aliyun.com/pypi/simple/   --trusted-host mirrors.aliyun.com

如果要在 TensorFlow 2 中支持 GPU,你需要额外安装以下几个组件:

  1. CUDA(Compute Unified Device Architecture)

CUDA 是 NVIDIA 提供的一个并行计算平台和编程模型,它允许 TensorFlow 在 GPU 上运行。

下载地址:developer.nvidia.com/cuda-toolki…

  1. cuDNN(CUDA Deep Neural Network Library)

cuDNN 是 NVIDIA 提供的一个 GPU 加速的深度神经网络库,优化了卷积神经网络等深度学习任务的计算速度。

下载地址:developer.nvidia.com/cudnn

我们新建一个helloWorld.py测试类:

import tensorflow as tf
​
print("TensorFlow版本:", tf.__version__)
print("CPU可用:", tf.config.list_physical_devices('CPU'))

运行结果:

TensorFlow版本: 2.20.0
CPU可用: [PhysicalDevice(name='/physical_device:CPU:0', device_type='CPU')]

说明环境搭建成功。