RTX2070显卡、tensorflow-gpu、CUDA10安装

1.硬件安装

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

还是上显卡吧!

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

最后选择:

  • 凄惨红 iGame GeForce RTX 2070 Ultra OC

    《RTX2070显卡、tensorflow-gpu、CUDA10安装》

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

    《RTX2070显卡、tensorflow-gpu、CUDA10安装》

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

    《RTX2070显卡、tensorflow-gpu、CUDA10安装》

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

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

    《RTX2070显卡、tensorflow-gpu、CUDA10安装》 图片发自简书App

东西到齐开始安装。。。

  • 原机箱

    《RTX2070显卡、tensorflow-gpu、CUDA10安装》
    《RTX2070显卡、tensorflow-gpu、CUDA10安装》

  • 显卡、电源到货

    《RTX2070显卡、tensorflow-gpu、CUDA10安装》

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

    《RTX2070显卡、tensorflow-gpu、CUDA10安装》
    《RTX2070显卡、tensorflow-gpu、CUDA10安装》
    《RTX2070显卡、tensorflow-gpu、CUDA10安装》
    《RTX2070显卡、tensorflow-gpu、CUDA10安装》

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

  • 安装完成!

    《RTX2070显卡、tensorflow-gpu、CUDA10安装》

2.软件安装

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

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

    《RTX2070显卡、tensorflow-gpu、CUDA10安装》

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

    《RTX2070显卡、tensorflow-gpu、CUDA10安装》
    《RTX2070显卡、tensorflow-gpu、CUDA10安装》

  • 安装CUDA 9.2

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

  • 安装CUDA 10

    《RTX2070显卡、tensorflow-gpu、CUDA10安装》

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

    《RTX2070显卡、tensorflow-gpu、CUDA10安装》

  • 安装 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啊

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

pip install tensorflow_gpu-1.11.0-CP36-cp36m-win_amd64.whl
  • 运行 import ...,成功!

以上, 全部搞定!!

    原文作者:shenbo
    原文地址: https://www.jianshu.com/p/7a8fcd11d1f6
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞