问题背景:以前用自己的电脑,小小的cpu,跑个mnist,能跑几十分钟(衰)。现在有了块显卡,虽然也不咋样,GTX1050,不过好歹也要试试GPU呀。于是参考着各位大佬的教程,一步步摸索,也自己记录一下,出份教程来。
本文参考链接:
https://blog.csdn.net/qq_15969343/article/details/79971469
https://www.cnblogs.com/elroye/p/7864988.html
https://blog.csdn.net/w401229755/article/details/78458014
https://blog.csdn.net/shuiyuejihua/article/details/78738664
https://blog.csdn.net/m0_37160535/article/details/80077903
一、安装Anaconda以及用Anaconda安装tf
这个,我之前也写过一篇博文
https://blog.csdn.net/qq_28123095/article/details/78811598
只不过安装的是tensorflow-gpu
pip install tensorflow-gpu
另,tensorflow已经支持python3.6了,不用创建3.5的环境来弄了。但我这次用的是3.5
activate tensorflow后,进入python里:
出现问题:
没有cuda 9.0,那就装咯~进入下一步
ps:后来因为要和其他同学版本一致,卸载了9.0,重装了8.0,据说卸载cuda很麻烦,但我觉得还好。参考链接:
https://blog.csdn.net/shuiyuejihua/article/details/78738664
二、安装CUDA
到官网,按你自己的需求选择合适的版本,下载。官网地址:
https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=7&target_type=exelocal
这个是9.0的,其他版本的就下其他版本的。
下载好后就安装嘛。结果就出现问题了:
???
我都让你装我C盘了,你还失败?
然后就百度了一下,度娘说有几种问题:
1)开了电脑管家、360之类的软件
2)驱动不匹配,建议用鲁大师装驱动
搞定这两点,就安装好啦
对了,我再安装过程中还出现了黑屏。。。但是我关机重启后就好了,也不知道是为什么???
然后在cmd窗口输入
nvcc -V
出现cuda版本信息,证明没问题
三、安装Cudnn
当然还是官网下载啦,链接:
https://developer.nvidia.com/rdp/form/cudnn-download-survey
需要注册/登录才行,那就注册一个账号咯,然后下载适配CUDA的版本
下载解压后放到c盘中:
配一下环境变量:
把”c:\cuda\bin”加入path
四、测试
在cmd中输入下列语句
成功!!!