Win10+TensorFlow+CUDA+cuDNN

先写上日期2018年8月24日。超过3个月就不要照着做了。

0x01 安装python3.6.6

官网下载,没啥说的。

0x02 安装CUDA

CUDA Toolkit 9.2 Download

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

大功告成。

    原文作者:于小葵
    原文地址: https://zhuanlan.zhihu.com/p/42845642
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞