分类
木讷技术宅

如何在Anaconda安装Tensorflow 2.0

年更选手赶在年底前更新一波。

肉壳之前在写神经网络相关的程序时,大多用的是Keras,因为组里的学长在用,而且我的项目也是基于一组Keras的代码。最近由于强迫症发作,在重装系统,发现Tensorflow已经升级到2.0了,而且也早就集成了Keras了,所以干脆想要一步到位升级到TF 2.0就算了。但是由于TF 2.0现在还只是测试版,所以Anaconda没有集成好的安装包,所以需要自己手动安装一些库,好在还是比较简单的。

  1. 安装Anaconda就不多说了:https://www.anaconda.com/distribution/
  2. 创建一个新环境
    conda create -n your_env_name python=3.7 cudnn cupti cudatoolkit=10.0
  3. 激活新环境
    conda activate your_env_name
  4. 安装Tensorflow-gpu 2.0 rc1
    pip install tensorflow-gpu==2.0.0-rc1
    pip install tensorflow-gpu

本篇文章写于19年9月16日,所以目前Anaconda还不能一键安装所有package,未来可能就可以了,另外,目前最新的TF版本是2.0.0-rc1,大家可以通过这里查询最新的2.0测试版版本号。还有,如果你安装rc版本的话,有可能在使用pycharm的时候,会遇到无法解析package的问题。官方说可能要等到2.0正式版出了。原因详见这个issue。如果有强迫症的话,安装beta版本就可以了,貌似这个问题是从rc版本开始有的。

19年10月4日更新,TF2.0正式版已出了,所以已经更新了安装命令了。另外,此文章不再更新cudatoolkit的安装版本,如果你在很久以后看到这篇文章,记得查一查cudatoolkit的版本,麻烦了。

大概就是这样……