Win10+RTX2060安装TensorFlow+Keras

Win10+RTX2060安装TensorFlow+Keras

Author:Gary
Date:2019-6-8
参考资料:https://blog.csdn.net/qq_32728345/article/details/81570504

主要按照参考资料中的内容进行安装，这里对其中的某些坑进行补充

1.安装显卡驱动

英伟达官网下载驱动安装即可，但是下载驱动后一定要在Nvidia控制面板中看一下支持的CUDA版本，目前最新版的驱动仅支持CUDA10.2，但是CUDA的下载页面中并没有CUDA10.2，不知道什么情况。这种情况下需要下载老版本的驱动，我一个一个试，降了2个版本后发现支持10.1的最新驱动版本为430.39，直接覆盖安装即可。

2.下载CUDA

直接进CUDA官网 https://developer.nvidia.com/cuda-toolkit-archive 下载对应的CUDA版本安装即可，没什么问题。然后要按照参考资料的步骤设置环境变量，注意把有版本号的地方换成自己的CUDA版本。

3.下载cuDNN

去https://developer.nvidia.com/rdp/cudnn-archive 下载对应CUDA版本的，解压后按参考资料复制到CUDA的目录中即可。

3. 下载安装Python

相信看这个教程的大部分人都安了的，大家可以放心用新版本Python，有大神为高版本Python编译TensorFlow

4.下载TensorFlow

TensorFlow官方更新很慢，可以到Github中下载其他人编译的新版本，注意CUDA,cuDNN,Python版本要对应。地址：https://github.com/fo40225/tensorflow-windows-wheel

下载后解压得到.whl文件，使用指令pip install <xxx.whl>安装

安装完成后可以在Python命令行中运行import tensorflow,没有报错即安装成功

5. 安装Keras和Numpy

pip install numpy
pip install keras

觉得下载很慢的童鞋可以参考我的另一个博客，简单2行指令疯狂加速pip安装速度，只看方法2即可:
https://blog.csdn.net/m0_37340681/article/details/84109320

安装完成后运行import keras，提示使用TensorFlow后端即安装成功了

可以运行以下程序查看是否用的GPU:

import os
from tensorflow.python.client import device_lib
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "99"

if __name__ == "__main__":
    print(device_lib.list_local_devices())

6. 可以开始你牛逼的深度学习之旅了