上传者: 49151708
|
上传时间: 2022-11-25 15:27:10
|
文件大小: 258KB
|
文件类型: PDF
tensorflow gpu 使用 pip 安装 TensorFlow
TensorFlow 2 软件包现已推出
• tensorflow :支持 CPU 和
GPU
的最新稳定版(适用于 Ubuntu 和 Windows)
• tf-nightly :预览 build(不稳定)。Ubuntu 和 Windows 均包含
GPU 支持。
旧版 TensorFlow
对于 TensorFlow 1.x,CPU 和 GPU 软件包是分开的:
• tensorflow==1.15 :仅支持 CPU 的版本
• tensorflow-gpu==1.15 :
支持 GPU
的版本(适用于 Ubuntu 和 Windows)
系统要求
• Python 3.6–3.9
◦ 若要支持 Python 3.9,需要使用 TensorFlow 2.5 或更高版本。
◦ 若要支持 Python 3.8,需要使用 TensorFlow 2.2 或更高版本。
• pip 19.0 或更高版本(需要 manylinux2010 支持)
• Ubuntu 16.04 或更高版本(64 位)