Question

我正在尝试使用NiftyNet并在第一步遇到困难。尝试运行快速启动命令

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

给了我

KeyError：＆＃34;使用名称＆＃39; FloorMod＆＃39;注册两个渐变！（之前的注册是_find_and_load_unlocked：955）＆＃34;

有人可以帮忙吗？我正在使用带有Nvidia GPU的Ubuntu 16.04。尝试张量流：1.4.1-py3码头图片，Anaconda与CPU版本tensorflow 和本机python与CPU版本tensorflow，我得到相同的错误。

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

谢谢！

Answer 1

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

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

Answer 2

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

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

2 个答案: