Conda 安装会自动安装 CUDA 和 GPU 支持所需的 CuDNN 库,但 pip 安装需要你手动完成。大家都比较喜欢一步到位的吧,特别是下载很多个库的时候。
快速启动
使用 Conda 安装 tensorflow
pip uninstall tensorflow
如果还没安装 Anaconda 或 Miniconda,那就安装这两者其中一个。Miniconda 只安装 Conda 和它的环境依赖,而 Anaconda 会预先为你安装好很多包。我倾向于使用 Miniconda 着手。待 Conda 下载好后,输入以下命令:
conda install tensorflow
如果你想用 GPU 驱动的版本,就把 tensorflow 替换为 tensorflow-gpu。
参考:
https://zhuanlan.zhihu.com/p/46599887