Pycharm 配置 gpu版TensorFlow

Pycharm: 2019.3.1
Python: 3.7
cuda: v10.1
cudnn: v7.6.4
TensorFlow: 2.0

最近重装了系统，打算重新配置tensorflow的环境:D

1.cuda 下载

本来是直接在官网下载的10.2版本，突然发现cudnn没有支持10.2版本的，就下载了10.1版本,选择自己的系统适合的版本
下载网址：https://developer.nvidia.com/cuda-10.1-download-archive-update2

许可协议，同意并继续

自定义安装，也可以选择精简，我是因为重装了系统电脑里没有VS，需要把VS依赖选项勾掉，不然会出错，有VS并且版本正确的可跳过这一步


默认装C盘，下一步


查看，在cmd输入

nvcc --version

2.cudnn下载

下载需要登陆账号！
下载链接https://developer.nvidia.com/rdp/cudnn-archive#a-collapse51b

将文件解压缩，复制这三个文件夹到cuda根目录下

3.pycharm 配置

点击File->setting->Project Interpreter，点击+ ,搜索tensorflow,下载tensorflow-gpu包

测试：

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))