更新后Tensorflow Extreme变慢

时间:2019-01-17 11:36:27

标签: python tensorflow ubuntu-18.04

我最近跑步了:

  • 获取更新
  • 获得升级

在Ubuntu 18.04上。我注意到它升级了一些与nvidia相关的软件包。 升级后,tensorflow大大减慢了速度。在升级培训之前,测试网络需要75秒,而现在大约需要15分钟。

我的版本:

  • cuda 10.0
  • nvidia驱动程序415.27
  • Cuda编译工具版本9.1,V9.1.85

在tensorflow conda env中:

  • cudatoolkit 9.2
  • cudnn 7.2.1
  • python 3.6.8
  • tensorflow / tensorflow-base / tensorflow-gpu 1.12.0

我已经尝试了许多方法来解决此问题,包括仅用于tensorflow的新conda环境,其他gpu驱动程序(390、410),重新安装gpu驱动程序。

我不知道如何找到问题的根源。我正在使用gtx 1080ti。我可以运行某种基准测试吗?

我尝试运行tensorflow cnn benchmark,但这需要tf_nightly_gpu,它尚不支持cuda 10.0。

0 个答案:

没有答案