RTX2070显卡、tensorflow-gpu、CUDA10安装

发表于 分类于 阅读次数： 本文字数： 575 阅读时长 ≈ 2 分钟

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/install/gpu。

  • 注意只支持到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安装。

    1
    pip install tensorflow-gpu

  • 赶紧 import 试一下：

    1
    2
    3
    import tensorflow as tf

    print(tf.__version__)

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

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

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

以上， 全部搞定！！