Tensorflow1.x-gpu环境下安装Tensorflow2.0-gpu
作者头像
  • 机器人网mp
  • 2019-11-10 14:37:46 9

最近,有不少朋友询问如何安装TensorFlow 2.0。我的项目目前主要使用的是1.x版本,但为了适应未来的发展,决定尝试安装TensorFlow 2.0。

鉴于我已经配置好了CUDA和cuDNN环境,所以不需要再次安装CUDA。对于那些还没有安装CUDA的用户,可以参考我之前发布的《Windows 10安装最新TensorFlow GPU版本指南》。该指南会自动安装对应版本的CUDA,无需手动操作。因此,对于尚未安装CUDA的用户,建议参照该指南;而对于已经安装CUDA或仅需CPU版本的用户,可以跳过这一步。

接下来,我们直接进入安装步骤:

  1. 打开Anaconda Prompt,在命令行中创建一个新的环境名为tf2,并指定Python版本为3.7。

    shell conda create -n tf2 python=3.7

  2. 激活刚刚创建的tf2环境。

    shell activate tf2

  3. 如果你还没有安装CUDA,可以执行以下命令来安装CUDA工具包和cuDNN库。如果已经安装了,则可以跳过这一步。

    shell conda install cudatoolkit=10 cudnn=7.6

  4. 最后,安装TensorFlow 2.0版本。安装过程中会自动安装所需的其他依赖包。此版本于10月25日发布,属于稳定版。

    shell conda install tensorflow-gpu=2.0.0

按照以上步骤操作,安装过程相对简单。安装完成后,可以看到安装成功的提示。

这样,你就完成了TensorFlow 2.0的安装。希望这些信息对你有所帮助。

    本文来源:图灵汇
责任编辑: : 机器人网mp
声明:本文系图灵汇原创稿件,版权属图灵汇所有,未经授权不得转载,已经协议授权的媒体下载使用时须注明"稿件来源:图灵汇",违者将依法追究责任。
    分享
Tensorflowgpu安装环境2.0
    下一篇