复现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