本文只对配置win10系统下tensorflow的GPU版本的流程作一记录,详细信息请参考其他网上教程。
1、安装python
可以只安装python,或安装anaconda,由于anaconda中附带了好多库,建议安装anaconda,方便以后的使用。
anaconda各个历史版本软件在清华镜像网站有,根据需要下载。
anaconda清华镜像站下载
2、安装cuda
cuda的安装注意需要在Nvidia控制面板中查看本机显卡是否支持
如上图所示,本机支持的cuda版本为cuda9.2.148,cuda版本必须和硬件设备保持一致。
3、安装cudnn
安装cuda9.2.148所对应的cudnn7.2.1.38
4、安装tensorflow
一般可采用pip install tensorflow-gpu==1.9.0 来完成安装过程,但是由于本机只支持cuda9.2,而tensorflow目前只支持cuda9.0,因此安装github上大神编译的tensorflow版本,完美解决这一问题。
tensorflow_gpu-1.9.0-cp36-cp36m-win_amd64
链接:https://pan.baidu.com/s/1yHUFaRaloI55GDKdoKqnTA 提取码:itzk
如果需要以上软件,可留言索取。
win10+GTX1060+cuda9.2.148+cudnn7.2.1.38+tensorflow1.9.0
最新推荐文章于 2019-03-25 23:34:09 发布