先写上日期2018年8月24日。超过3个月就不要照着做了。
0x01 安装python3.6.6
官网下载,没啥说的。
0x02 安装CUDA
0x03 安装cuDNN
cuDNN 7.2
https://developer.nvidia.com/rdp/form/cudnn-download-survey
这个需要自己注册一下,然后登陆下载。
将下载解压后的三个文件复制到CUDA相应位置。
具体位置自己百度去找。
0x04 安装tensorflow-gpu
默认pip安装的是1.10 但是不支持CUDA 9.2。只支持9.0.于是找网上现成的装一下。
https://github.com/fo40225/tensorflow-windows-wheel
恩,基础环境部署完成。为了方便使用tf,建议学习tflearn
0x05 安装tflearn
pip install tflearn
测试一下成果吧。
git clone https://github.com/tflearn/tflearn
进examples/images目录
python alexnet.py
大功告成。