【解决错误】NameError: name ‘Pipeline‘ is not defined -- nvidia.DALI

错误描述

Please install DALI from https://www.github.com/NVIDIA/DALI to run this example.
Traceback (most recent call last):
  File "imagenet.py", line 28, in <module>
    from utils.dataloaders import *
  File "/home/project/mobilenetv2.pytorch/utils/dataloaders.py", line 19, in <module>
    class HybridTrainPipe(Pipeline):
NameError: name 'Pipeline' is not defined

原因分析

• 基于错误提示信息以及回溯，经过分析发现是少了Nvidia的DALI依赖包。

解决方法

• 使用pip安装
• 首先查看cuda版本以及python版本，安装相应版本的DALI依赖包
  • 查看cuda版本：nvcc -V
  • 查看python版本：python/python3 -V
• pip安装命令：pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/ nvidia-dali-cuda102==1.6.0