TENSORFLOW 2.0 发布版已经发布!
自 2019 年 1 月 10 日起,我说的不是测试版,而是发布版。
使用 Anaconda
自 2019 年 1 月 11 日起,Anaconda 开始支持 Tensorflow 2.0.0。
选项 1: 最简单的方法是:
conda install tensorflow 或 conda install tensorflow-gpu
对于 gpu 模式,anaconda 会处理所有你需要安装的 CUDA 以使 tensorflow gpu 模式工作,所以我强烈建议使用这种方法。
这种方法的唯一问题是 anaconda 可能没有最新的 TensorFlow 版本。例如,在 2021 年 2 月 21 日,conda 的版本是 2.3,而 PIP 版本是 2.4。您可以查看gpu或cpu的当前版本。
选项 2(虚拟环境): 使用环境安装 tensorflow 是strongly recommended,为此您需要以下命令先创建环境,然后在其中安装 tensorflow :
- CPU:
conda create -n <your_env_name> tensorflow
-
GPU:
conda create -n <your_env_name> tensorflow-gpu
将<your_env_name> 更改为有意义的名称,例如tf-2
使用 tensorflow 先运行conda activate <your_env_name>
使用点子
使用 pip 的 tensorflow official instructions 相当完整。
只需使用 pip 安装 tensorflow,例如:
# Current stable release for CPU-only
pip install tensorflow
我仍然建议在做任何事情之前在新环境中安装 tensorflow,这样 3 个步骤将是(使用 anaconda):
conda create --n <our_env_name> pip
conda activate <your_env_name>
pip install tensorflow
现在对于 GPU 版本,使用 pip 更加困难,我向您推荐这个 link,它解释了您需要安装的额外内容(CUDA 和其他)。