时间戳

首页 > 解决方案 > ModuleNotFoundError：运行 nvidia/deeplearningexamples reponstory 时没有名为“nvidia.dali.backend_impl”的模块

python - ModuleNotFoundError：运行 nvidia/deeplearningexamples reponstory 时没有名为“nvidia.dali.backend_impl”的模块

问题描述

在 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

标签： pythonnvidia

解决方案

ok，我已经解决了，因为NVIDIA没有对具体安装进行分类，其实不用从头安装，如果你用pip install，只要看看NVIDIA docker用的是哪个版本，然后下载这个具体的版本，pip install，一切都会好起来的。

推荐阅读