【问题解决】ModuleNotFoundError: No module named ‘nvidia‘：

复现Ultra-Fast-Lane-Detection-v2道路标线检测
具体参考：解决报错ModuleNotFoundError: No module named ‘nvidia‘：
报错：

ModuleNotFoundError: No module named 'nvidia'：

Traceback (most recent call last):
  File "train.py", line 8, in <module>
    from utils.common import calc_loss, get_model, get_train_loader, inference, merge_config, save_model, cp_projects
  File "/line_detection/UFLD-v2/utils/common.py", line 2, in <module>
    from data.dali_data import TrainCollect
  File "/line_detection/UFLD-v2/data/dali_data.py", line 4, in <module>
    from nvidia.dali.pipeline import Pipeline
ModuleNotFoundError: No module named 'nvidia'

	}

解决：
执行如下命令，即可：（要注意cuda版本对应，我的cuda是11.8使用11系列即可）

pip install --extra-index-url https://developer.download.nvidia.com/compute/redist --upgrade nvidia-dali-cuda110