用sklearn学习完了机器学习之后,本来打算入坑深度学习。查阅了网上相关资料之后决定学习tensorflow进行入门,哪知安装就花费了一周的时间。深感老显卡安装不易,在此记录一下安装过程希望可以帮助其他人少被坑。
本机机器配置:六代i5处理器、GTX965M显卡 、8G*2 DDRL3内存条、240G固态。
一、检查适合自己硬件的CUDA和tensorflow组件
首先确认显卡的驱动安装好了。WIN10环境进入英伟达控制面板,只需在桌面右击就行。这一步出现异常建议重新安装驱动或者自行百度解决。
进入控制面板->帮助->系统信息->组件->NVCUDA->产品名称,不多赘述直接上图。
买不起RTX2080只能用CUDA8.0了,确定了CUDA版本剩下的组件按照下表进行选择:
我习惯使用python3.6,tensorflow版本感觉越高功能可能齐全。所以在win10系统下决定用CUDA8.0+tensorflow1.4(GPU)+CUDNNv6.1+python3.6。编译器选择了VS 。
组件下载链接:
各版本CUDA下载链接
各版本CUDNN下载链接
各版本tensorflow(GPU)下载链接
二、
参考博文:
1.检查显卡支持哪个版本的CUDA
2.tensorflow CUDA cudnn 版本对应关系
3.