ModuleNotFoundError:运行nvidia/deeplearningexamples响应故事时没有名为'nvidia.dali.backend_impl‘的模块

Question

在NVIDIA/DeepLearningExamples中运行rn50脚本，如

./rn50_partial.sh

获取如下错误：

Traceback (most recent call last):

  File "/export/nfs/sunxue/DeepLearningExamples/PyTorch/Classification/ConvNets/mnasnet/training/FP32/../../../launch.py", line 7, in <module>

    from main import main, add_parser_arguments, available_models

  File "/export/nfs/sunxue/DeepLearningExamples/PyTorch/Classification/ConvNets/main.py", line 49, in <module>

    from image_classification.dataloaders import *

  File "/export/nfs/sunxue/DeepLearningExamples/PyTorch/Classification/ConvNets/image_classification/dataloaders.py", line 79, in <module>

    class HybridTrainPipe(Pipeline):

NameError: name 'Pipeline' is not defined

Answer 1

好的，我已经解决了这个问题，因为NVIDIA没有对具体的安装进行分类，实际上，它不需要从头开始安装，如果你使用pip install，只需检查NVIDIA docker使用哪个版本，并下载这个特定的版本，pip install，一切都会好的。