小编San*_*nyo的帖子

没有 NCCL 的镜像策略

  • 我是否编写了自定义代码(而不是使用 TensorFlow 中提供的股票示例脚本):否
  • 操作系统平台和发行版(例如 Linux Ubuntu 16.04):Windows 10 x64
  • TensorFlow 从(源代码或二进制文件)安装:二进制文件
  • TensorFlow 版本(使用下面的命令):1.8.0
  • Python 版本:3.6
  • Bazel 版本(如果从源代码编译):-
  • GCC/编译器版本(如果从源代码编译):-
  • CUDA/cuDNN 版本:9.0
  • GPU 型号和内存:3.5
  • 要重现的确切命令:simple_tfkeras_example.py

我想使用 MirroredStrategy 在同一台机器上使用多个 GPU。我尝试了以下示例之一:https : //github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/distribute/python/examples/simple_tfkeras_example.py

结果是:ValueError: Op type not register 'NcclAllReduce' in binary running on RAID。确保 Op 和 Kernel 已在此过程中运行的二进制文件中注册。在构建 NodeDef 'NcclAllReduce' 时

我使用的是 Windows,因此 Nccl 不可用。是否可以强制 TensorFlow 不使用此库?

tensorflow

5
推荐指数
1
解决办法
1962
查看次数

标签 统计

tensorflow ×1