Anaconda Tensorflow Keras 安装方法

前言

网上扒论文代码的时候，首先得先确定代码的效果，如果效果比较好的话，才考虑继续集成到自己的项目中。
但是经常会遇到很多开源代码用的是Tensorflow Keras等框架，不是我常用的Pytorch，直接移植代价还是比较高的，所以还是需要部署一下相关的环境用来初步测试。

配置步骤

1. 确定版本号

首先找到需要部署的项目中的README.md或者requirements.txt，找到需要的版本号：

Python 3.6.10
TensorFlow-gpu 1.14.0
Keras-gpu 2.2.4

但是不能全信，有些作者给的版本对应是错的，所以安装之前，最好搜一下作者给的版本对应是不是合理的

除了Python、TensorFlow-gpu、Keras-gpu版本外，还需要确定cudatoolkit和cudnn版本，可以根据以下链接查询

https://tensorflow.google.cn/install/source?hl=zh-cn#tested_build_configurations

英文官网才有最新版本对应，如果找不到需要的版本，请将语言切换为英文

cudatoolkit 10.0
cudnn 7.4

注意，不支持10.1

2. 使用Anaconda创建虚拟环境

创建的时候指定Python版本

conda create -n keras_gpu_224 python=3.6.10

进入虚拟环境

conda activate keras_gpu_224

3. 安装cudatoolkit与cudnn

conda install cudatoolkit=10 -c http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/
conda install cudnn=7 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/

4. 使用清华源安装Tensorflow

指定安装GPU版本、及版本号的Tensorflow

pip install tensorflow-gpu==1.14.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

5. 使用清华源安装Keras

指定版本安装Keras

pip install keras==2.2.4 -i https://pypi.tuna.tsinghua.edu.cn/simple