在TensorFlow开发中,创建一个独立的虚拟环境是推荐的做法,这样可以避免不同项目之间的依赖冲突。以下是使用 Python 的 virtualenv 工具来创建 TensorFlow 虚拟环境的步骤:

系统要求

  • Python 3.5 或更高版本
  • virtualenv 工具

创建虚拟环境

  1. 安装 virtualenv:

    pip install virtualenv
    
  2. 创建虚拟环境:

    virtualenv my_tensorflow_env
    

    这将在当前目录下创建一个名为 my_tensorflow_env 的虚拟环境。

  3. 激活虚拟环境:

    • 在 Windows 上:
      my_tensorflow_env\Scripts\activate
      
    • 在 macOS/Linux 上:
      source my_tensorflow_env/bin/activate
      

    激活后,命令行提示符会变成绿色,表示虚拟环境已激活。

安装 TensorFlow

在虚拟环境中,使用以下命令安装 TensorFlow:

pip install tensorflow

注意

  • 如果你需要安装 TensorFlow GPU 版本,请使用以下命令:

    pip install tensorflow-gpu
    

验证安装

安装完成后,可以通过以下命令验证 TensorFlow 是否安装成功:

import tensorflow as tf
print(tf.__version__)

扩展阅读

想了解更多关于 TensorFlow 的信息,请访问我们的 TensorFlow 教程页面

[center]TensorFlow_Logo