Windows安装TensorFlow 2.0 GPU正式版

安装步骤

目前anaconda还不支持一键安装tf2.0，所以需要“pip install” tensorflow ， “conda install” cuda和cudnn。

TensorFlow 2.0 需要cuda10.0，cudnn(>= 7.4.1)，同时对显卡驱动版本也有要求(>= 410.x)

【如果官方镜像速度慢，请更换国内镜像。】

换conda源

打开cmd，执行以下命令：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

如果需要换回默认源：

conda config --remove-key channels

路径C:/Users(或用户)/(你的用户名)/ 下会生成配置文件.condarc，可以用文本编辑工具打开查看。

换pip源

参考这篇文章：windows环境下永久修改pip镜像源的方法

具体步骤

具体步骤如下：（我是直接在Anaconda Prompt中操作的，cmd或powershell中没试过）

1. 创建新环境：

conda create -n tf2 pip

2. **新环境tf2

activate tf2

3. 安装tensorflow

pip install tensorflow-gpu

默认安装的就是2.0版本

4. 安装cuda

conda install cudatoolkit=10.0

默认安装会安装10.1版本，tf2暂时不支持10.1，所以请指定安装10.0的版本

5. 安装cudnn

conda install cudnn

默认安装最新版，满足版本要求；也可指定版本

6. 测试是否安装成功（此时仍在tf2环境中，依次键入以下命令）

python
import tensorflow

如果没有任何报错，说明已经安装成功。

其他问题

1. 如果你不小心装了cuda10.1，会遇到如下的问题，可以试试这样解决（解决方案后续代价暂且未知，最好还是重新安装cuda10.0）。
   解决方案：找到tf2环境的文件夹，一般在C:/Users(或用户)/(你的用户名)/Anaconda3/envs/tf2。然后搜索“cudart”，会出现一个“cudart64_101.dll”的文件，重命名为“cudart64_100.dll”，重新测试。