3

我正在尝试 NiftyNet 并在第一步卡住了。尝试运行快速启动命令

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

给我

KeyError：“注册两个名为‘FloorMod’的渐变！（之前注册在_find_and_load_unlocked:955）”

有人可以帮忙吗？我正在使用带有 Nvidia GPU 的 Ubuntu 16.04。尝试了 tensorflow:1.4.1-py3 docker 映像、带有 CPU 版本的 tensorflow 的 Anaconda 和带有 CPU 版本的 tensorflow 的本机 python，我得到了同样的错误。

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

谢谢！

4

2 回答 2

2

我认为这是从 TensorFlow 1.3（这是官方 NiftyNet 依赖项）到 1.4 的变化。可以切换回 TF1.3，或者在 niftynet/layer/resampler.py 中注释掉以下几行

@tf.RegisterGradient('FloorMod')
def _floormod_grad(op, grad):
    return [None, None]
于 2018-01-28T20:39:39.773 回答
0

请使用此命令安装 tensorflow pip install tensorflow 然后使用以下命令安装 nifty net ''' pip install niftynet ''' 重新安装 python ''' pip install python ''' 如果问题仍然存在，请在更多信息中提及详细信息请确保在从 niftynet 页面执行命令之前设置您的环境变量。

于 2019-11-13T06:01:43.107 回答