下载安装之前
首先,你要知道你需要下什么。
网速一定要够,不然真的很痛苦,特别是如果下载错了的话,很浪费时间
所以建议先阅读
各种依赖
系统:windows(本文)
tensorflow:深度学习框架 -gpu 使用gpu版本
cuda:用来支持gpu进行复杂计算
cudnn:嵌入cuda的一个神经网络加速库文件
nvidia驱动:用来驱动pc的nvidia显卡
关键点
在于版本依赖
-
检查显卡
你要查看自己的PC是否是有nvidia显卡的,没有的话你可以放弃了你要查看自己的PC是否是有nvidia显卡的,没有的话你可以放弃了 -
确定nvidia驱动版本
在控制面板里面查找nvidia,点进去找找有显卡适合的驱动
去官网下载,官网也会有针对不同显卡的搜索支持 -
下载CUDA,和CUDNN, Tensorflow要版本配对
可以先确定你要安装的tensorflow,建议不要安装最新的,往前一两个版本(由于时效性,具体说了也没用)
然后确定对应的CUDA 和 Cudnn
具体的版本对应网上很多,找一找下载完记得添加环境变量
-
tensorflow