尝试运行NiftyNet快速启动命令时出错

Question

我正在试用NiftyNet，在第一步就被困住了。试图运行快速启动命令

~/niftynet/extensions/dense_vnet_abdominal_ct/config.ini net_download.py dense_vnet_abdominal_ct_model_zoo python net_segment.py推理-c -c

给我

KeyError：“注册两个渐变名为'FloorMod‘！(以前注册是在_find_and_load_unlocked :955中)。”

有人能帮忙吗？我正在使用Ubuntu16.04与Nvidia GPU。尝试tensorflow:1.4.1-py3对接者映像，Anaconda与CPU版本的tensorflow和原生python与CPU版本的tensorflow，我得到同样的错误。

我很确定我做错了什么，因为我从不同的环境中得到了同样的错误，但我不知道是什么.

谢谢!

Answer 1

我认为这是从TensorFlow 1.3 (官方的NiftyNet依赖项)改为1.4。您可以切换回TF1.3，也可以在niftynet/layer.py中注释掉以下行

@tf.RegisterGradient('FloorMod')
def _floormod_grad(op, grad):
    return [None, None]

Answer 2

请使用此命令安装tensorflow，安装tensorflow之后，请使用下面的命令‘’pip install niftynet‘重新安装python’‘pip install python’如果问题还没有详细说明您的问题，请确保在从niftynet页面执行命令之前设置了环境变量。