Ubuntu 18.04 latest, RTX2080ti, tlt-streamanalytics:v2.0_dp_py2 container
I receive the error below (no module named nvml) when I run the tlt-dataset-convert command in the detectnet_v2 example. Deepstream 5 works fine on this box outside the container.
print(“Converting Tfrecords for kitti trainval dataset”)
!tlt-dataset-convert -d $SPECS_DIR/detectnet_v2_tfrecords_kitti_trainval.txt
-o $DATA_DOWNLOAD_DIR/tfrecords/kitti_trainval/kitti_trainval
************Output **************
Creating a new directory for the output tfrecords dump.
print(“Converting Tfrecords for kitti trainval dataset”)
!tlt-dataset-convert -d $SPECS_DIR/detectnet_v2_tfrecords_kitti_trainval.txt \
-o $DATA_DOWNLOAD_DIR/tfrecords/kitti_trainval/kitti_trainval
Converting Tfrecords for kitti trainval dataset
Using TensorFlow backend.
Traceback (most recent call last):
File “/usr/local/bin/tlt-dataset-convert”, line 5, in
from iva.detectnet_v2.scripts.dataset_convert import main
File “./detectnet_v2/scripts/dataset_convert.py”, line 14, in
File “./detectnet_v2/dataio/build_converter.py”, line 13, in
File “./detectnet_v2/dataio/kitti_converter_lib.py”, line 22, in
File “./detectnet_v2/dataio/dataset_converter_lib.py”, line 20, in
File “./detectnet_v2/dataloader/utilities.py”, line 15, in
File “/usr/local/lib/python2.7/dist-packages/modulus/init.py”, line 8, in
from modulus import blocks
File “/usr/local/lib/python2.7/dist-packages/modulus/blocks/init.py”, line 22, in
from modulus.blocks import data_loaders
File “/usr/local/lib/python2.7/dist-packages/modulus/blocks/data_loaders/init.py”, line 9, in
from modulus.blocks.data_loaders.sqlite_dataloader import SQLiteDataLoader
File “./modulus/blocks/data_loaders/sqlite_dataloader.py”, line 10, in
File “/usr/local/lib/python2.7/dist-packages/modulus/dataloader/init.py”, line 8, in
from modulus.dataloader import humanloop
File “./modulus/dataloader/humanloop.py”, line 16, in
File “/usr/local/lib/python2.7/dist-packages/modulus/processors/init.py”, line 26, in
from modulus.processors.buffers import NamedTupleStagingArea
File “./modulus/processors/buffers.py”, line 10, in
File “/usr/local/lib/python2.7/dist-packages/modulus/hooks/init.py”, line 9, in
from modulus.hooks.hooks import KerasCheckpointListener
File “./modulus/hooks/hooks.py”, line 26, in
ImportError: No module named nvml
Thanks,
Patrick