Open
Description
🐛 Describe the bug
As part of the DALI integration PR, #1958,
A pytest test_dali_preprocess was added.
This expects nvidia-dali-cuda110==1.18.0
Regression Tests error if this is not installed.
Error logs
==================================== ERRORS ====================================
_____________ ERROR collecting test/pytest/test_dali_preprocess.py _____________
ImportError while importing test module '/home/runner/work/serve/serve/test/pytest/test_dali_preprocess.py'.
Hint: make sure your test modules/packages have valid Python names.
Traceback:
/opt/hostedtoolcache/Python/3.8.16/x64/lib/python3.8/importlib/__init__.py:127: in import_module
return _bootstrap._gcd_import(name[level:], package, level)
test_dali_preprocess.py:10: in <module>
from examples.nvidia_dali.custom_handler import DALIHandler
../../examples/nvidia_dali/custom_handler.py:10: in <module>
from nvidia.dali.pipeline import Pipeline
E ModuleNotFoundError: No module named 'nvidia'
=========================== short test summary info ============================
ERROR test_dali_preprocess.py
!!!!!!!!!!!!!!!!!!!! Interrupted: 1 error during collection !!!!!!!!!!!!!!!!!!!!
=============================== 1 error in 4.02s ===============================
Installation instructions
Run regression tests
Model Packaing
NA
config.properties
NA
Versions
------------------------------------------------------------------------------------------
Environment headers
------------------------------------------------------------------------------------------
Torchserve branch:
torchserve==0.7.1b20230323
torch-model-archiver==0.7.1b20230323
Python version: 3.8 (64-bit runtime)
Python executable: /home/ubuntu/anaconda3/envs/torchserve/bin/python
Versions of relevant python libraries:
captum==0.5.0
intel-extension-for-pytorch==1.13.0
numpy==1.23.4
nvgpu==0.9.0
psutil==5.9.4
pygit2==1.6.1
pylint==2.6.0
pytest==7.2.0
pytest-cov==4.0.0
pytest-mock==3.10.0
pytorchvideo==0.1.5
requests==2.28.1
requests-futures==1.0.0
requests-toolbelt==0.9.1
torch==2.0.0+cpu
torch-model-archiver==0.7.1b20230323
torch-model-archiver-nightly==2022.7.12
torch-workflow-archiver==0.2.7b20230323
torch-workflow-archiver-ag==0.2.4b20220624
torch-workflow-archiver-nightly==2022.7.12
torchaudio==2.0.1+cpu
torchdata==0.6.0
torchmetrics==0.9.3
torchmultimodal-nightly==2022.10.27
torchrec==0.2.0
torchserve==0.7.1b20230323
torchserve-nightly==2022.11.8
torchserve-nightly-ag==2022.6.29
torchtext==0.15.1
torchvision==0.15.1+cpu
torchx-nightly==2022.8.16
transformers==4.11.0
wheel==0.40.0
torch==2.0.0+cpu
torchtext==0.15.1
torchvision==0.15.1+cpu
torchaudio==2.0.1+cpu
Java Version:
OS: Ubuntu 18.04.6 LTS
GCC version: (Ubuntu 7.5.0-3ubuntu1~18.04) 7.5.0
Clang version: N/A
CMake version: version 3.24.0
Repro instructions
Run regression test
Possible Solution
Exclude the test from regression tests