Windows10下Anaconda安装tensorflow2.1.0_tensorflow-gpu 2.1.0 对应的cuda版本

Windows10下Anaconda安装tensorflow2.1.0

笔者最近在安装tensorflow2.1.0，有一些心得，仅供大家参考。

首先是电脑环境，笔者电脑有
Anaconda3\Pycharm\Visual Studio 2019\CUDA 10.1\cuDNN10.1

Anaconda官网
https://www.anaconda.com/distribution
安装方法有很多可以自行搜索，我的建议就是安装步骤来就行，记住自己安装路径，安装完后，将目录添加到系统环境变量path就可以了。

Pycharm官网
https://www.jetbrains.com/pycharm/download/#section=windows
同样有很多安装教程，我的建议也是安装步骤来就行。

目前直接PIP安装最新tensorflow会直接GPU版本故先装CUDA和cuDNN

tensorflow2.1.0对应的是CUDA10.1，如果版本不对，可能会出现找不到cudart64_101.dll的情况

CUDA官网下载，选择Windows 64位系统 win10 local本地下载，
https://developer.nvidia.com/cuda-toolkit-archive

cuDNN官网下载，不过需要提前注册NVIDIA账号才行，选择对应的10.1下载
https://developer.nvidia.com/rdp/cudnn-download

双击运行cuda_10.1.243_426.00_win10，建议都默认设置来就可以了，一般默认系统环境变量path也会在安装中自动添加。
笔者电脑已经有visual studio，所以没有的人，可以在安装过程中点高级选项，把Visual Studio(VS) Intergration√取消。不过安装一个visual studio也并不麻烦，可以提前安装一个Visual Studio2017或者2019的社区版，反正免费。

解压cuDNN的压缩包，将cudnn-10.1-windows10-x64-v7.6.5.32\cuda下的三个文件夹bin\include\lib放入CUDA安装目录即安装完成，如果是默认，就在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1。

打开cmd，默认是在C:\Users\Administrator下，连续两次cd …/转到C:\然后cd转到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin，再输入nvcc -V，显示编号即成功。

接下来安装tensorflow

打开Anaconda Prompt，输入如下代码

conda create -n tensorflow python=3.7
activate tensorflow
pip install --upgrade --ignore-installed tensorflow -i https://pypi.doubanio.com/simple

1
2
3
4

这里用的是豆瓣镜像源，个人觉得最快的一个了。

安装完成后，打开Anaconda Prompt输入测试代码

activate tensorflow
python

import tensorflow as tf
print(tf.__version__)
print('GPU', tf.test.is_gpu_available())
a = tf.constant(2.0)
b = tf.constant(4.0)
print(a + b)
1
2
3
4
5
6
7
8
9

结果如下

这是Anaconda Navigator里面显示已安装环境:

这是Pycharm里面导入Anaconda tensorflow环境后测试：

完成！