RTX2070显卡、tensorflow-gpu、CUDA10安装

title: RTX2070显卡、tensorflow-gpu、CUDA10安装 date: 2018-12-29 22:50:50 tags:

• keras
• tensorflow
• tensorflow-gpu
• RTX2070
• CUDA10 categories: tensorflow

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线

  

东西到齐开始安装。。。

• 原机箱

  

  

• 显卡、电源到货

  

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

  

  

  

  

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

• 安装完成！

  

2.软件安装

• 参考Tensorflow官网的tensorflow-gpu版本,安装指南, https:// tensorflow.google.cn/in stall/gpu 。

  • NVIDIA® GPU 驱动程序, https://www. nvidia.com/drivers - CUDA 9.0 需要 384.x 或更高版本。
  • CUDA® 工具包, https:// developer.nvidia.com/cu da-zone - TensorFlow 支持 CUDA 9.0。
  • CUDA 工具包附带的 CUPTI, http:// docs.nvidia.com/cuda/cu pti/
  • cuDNN SDK, https:// developer.nvidia.com/cu dnn （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