下载安装之前
首先,你要知道你需要下什么。
网速一定要够,不然真的很痛苦,特别是如果下载错了的话,很浪费时间
所以建议先阅读
各种依赖
系统:windows(本文)
tensorflow:深度学习框架 -gpu 使用gpu版本
cuda:用来支持gpu进行复杂计算
cudnn:嵌入cuda的一个神经网络加速库文件
nvidia驱动:用来驱动pc的nvidia显卡
关键点
在于版本依赖
-
检查显卡
你要查看自己的PC是否是有nvidia显卡的,没有的话你可以放弃了你要查看自己的PC是否是有nvidia显卡的,没有的话你可以放弃了 -
确定nvidia驱动版本
在控制面板里面查找nvidia,点进去找找有显卡适合的驱动
去官网下载,官网也会有针对不同显卡的搜索支持 -
下载CUDA,和CUDNN, Tensorflow要版本配对
可以先确定你要安装的tensorflow,建议不要安装最新的,往前一两个版本(由于时效性,具体说了也没用)
然后确定对应的CUDA 和 Cudnn
具体的版本对应网上很多,找一找下载完记得添加环境变量
-
tensorflow安装
建议去官网,里面有教程,使用pip安装,记住不要急,一定要检查前面的是否正确下载了
否则会找不到dll模块的
当然安装cpu版本的就没这个问题了
其他
使用gpu的话用个人电脑还是很少
一般如果在公司或者实验室的话,会提供已经装好的服务器,这时候只需要安装
openssh 和 xshell就可以远程登录服务器跑代码了
如果只是想在本机试着跑一些简单程序,下载cpu版本就好了,问题少很多