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. 可以开始你牛逼的深度学习之旅了
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
