RTX2070显卡、tensorflow-gpu、CUDA10安

1.硬件安装

前一段用 tensorflow 玩深度学习，笔记本只能跑个 mnist，再大点的模型都跑不动。
试了腾讯云、极客云，用着也不方便。

还是上显卡吧！

• 本来一门心思想买1080，看着看着RTX20系列出了，买新不买旧，那就买2070吧。
• 学校有个台式机是dell optix 7020，小机箱，原装电源只有～300w。还得换个电源。
• 机械硬盘还得再换个ssd。。。

最后选择：

• 凄惨红 iGame GeForce RTX 2070 Ultra OC

  
  

• Intel 545S 256G固态硬盘 2.5英寸盘 SATA接口 SSD

  
  

• 安钛克 EAG550evo金牌全模组电源

  
  

这个台式机的主板供电接口不是通用的，坑爹，还得配一个转接头：

• DELL Optiplex 3020 7020 9020主板 8针电源线 电源24PIN转8PIN线

  
  RTX2070显卡、tensorflow-gpu、CUDA10安装

东西到齐开始安装。。。

• 原机箱

  
  
  

• 显卡、电源到货

  
  

• 原硬盘和支架挡到了显卡，果断拆掉。。。

  
  
  
  
  

事实证明，并不好拆。。。

• 安装完成！

  
  

2.软件安装

• 参考Tensorflow官网的tensorflow-gpu版本安装指南。

  • NVIDIA® GPU 驱动程序 - CUDA 9.0 需要 384.x 或更高版本。
  • CUDA® 工具包 - TensorFlow 支持 CUDA 9.0。
  • CUDA 工具包附带的 CUPTI。
  • cuDNN SDK（7.2 及更高版本）

• 注意只支持到CUDA 9，到处是坑。。。
  

  

• 首先，安装显卡驱动，Win10已经自动装好了。可以到nvidia官网确认一下驱动的版本是不是最新的。

  
  
  

• 安装CUDA 9.2

CUDA官网的最新版已经是CUDA10；找到历史版本试着安装CUDA9.2，但是试了几次都安装失败。。。
估计是CUDA9.2不支持RTX2070。。。
那就硬着头皮安装CUDA10吧。

• 安装CUDA 10

  
  

• 安装cuDNN 7.3.1，需要注册一个 nvidia developer账号。

  
  

• 安装 tensorflow-gpu，最新版是 tensorflow_gpu-1.11.0，直接用pip安装。

pip install tensorflow-gpu

• 赶紧 import 试一下：

import tensorflow as tf

print(tf.__version__)

但是运行时却报错Import Error: DLL load fail: 找不到指定的模块

阿西吧。。。看来真不支持CUDA10啊

最后请教论坛大神，找到了野生版轮子：

• tensorflow-windows-wheel
  

  

• 找到tensorflow_gpu-1.11.0-CP36-cp36m-win_amd64.whl
  

  

• 下载到本地安装

pip install tensorflow_gpu-1.11.0-CP36-cp36m-win_amd64.whl

• 运行 import ...，成功！

以上， 全部搞定！！