1903 重置 1708 版(就算笔记本是 965M,我也要搞 TF-GPU!)

原文地址:https://www.yuangezhizao.cn/articles/python/TensorFlow/install-tensorflow-gpu-on-Win-10.html
原文作者:远哥制造

0x00.缘由

2017-8-3 10:25:03写的Win-10 安装 TensorFlow-GPUCSDN上的一位博主引用,这篇文章让我这个小站有了每天的固定个位数访客。正值前几天TF 2.0发布(油管直播我他妈熬夜看的QAQ),也是时候把这篇文章更新下了,其实早就想了……
唔,还有简书上的这篇Faster R-CNN for Tensorflow

0x01.环境

a. Microsoft Windows [版本 10.0.17763.316](Win 10 x64 2019-LTSC 1809 17763.316)
b. Python 3.7.2 (tags/v3.7.2:9a3ffc0492, Dec 23 2018, 23:09:28) [MSC v.1916 64 bit (AMD64)] on win32
c. JetBrains Pycharm 2018.3 x64 Professional
d. GeForce GTX 965M

0x02.安装

写流水账开始了啊,具体的细节自己去翻那篇吧。

1. 去TF 官网

CUDA 9.0
CUDA 9.0

卧槽,然后我就被坑了,都下完了突然想到为啥不试试CUDA 10.0?可官网没写支持啊?况且在其他地方我也看到了说不支持的,XD……
结果忘了是要装NVIDIA的哪个包来着的,要求CUDA 10.0,WTF?然后就去谷歌搜了,结果还真搜到了,不过是谷歌快照……

CUDA 10.0
CUDA 10.0

当时就傻逼了,这他妈中英文文档内容居然不一样,中文的更新速度居然这么慢,原来官方都支持CUDA 10.0了啊?!

GPU support
GPU support
Pip package
Pip package
Windows setup
Windows setup

卧槽,切换按钮在页面最右下角,绝了!

Windows setup
Windows setup

2. 安装CUDA

算力要3.5以上才可以,我的是5.2

GeForce GTX 965M
GeForce GTX 965M
最新 10.1
最新 10.1
选择 10.0
选择 10.0
2.1 GB 真他妈大啊
2.1 GB 真他妈大啊

链接:https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda_10.0.130_411.31_win10

GD-WLAN 好慢呀
GD-WLAN 好慢呀

滚去下net版了(注:此处下错了,下最新的10.1了……已改正图)

这他妈叫精简,WTF?
这他妈叫精简,WTF?
真·最小化安装
真·最小化安装
打死也不升级
打死也不升级
PATH 环境变量已经给写好了
PATH 环境变量已经给写好了
装完之后拿 nvcc -V 验证
装完之后拿 nvcc -V 验证
确认无误
确认无误

补充:10.1的报错截图:

Fuck u 10.1
Fuck u 10.1

3. 安装cuDNN

下载 10.0
下载 10.0
主要部分还是这个 DLL,不要管截图中的 10.1,自己去下 10.0
主要部分还是这个 DLL,不要管截图中的 10.1,自己去下 10.0

4. 安装TensorRT 5.0

下载
下载
下载
下载
安装
安装

安装方法和cuDNN同理,只把/lib目录下的全部内容复制到/bin目录下嘛

这个最新的倒是只有 10.0
这个最新的倒是只有 10.0

5. 终于完事儿了

可海星
可海星

未完待续……