To narrow down, can you login the docker
docker run --runtime=nvidia -it -v
<yourfolder>
:/workspace/tlt-experiments nvcr.io/nvidia/tlt-streamanalytics:v3.0-dp-py3 /bin/bash
In the docker, please refer to python - How to remove cuInit failed: unknown error in CUDA (PyCuda) - Stack Overflow and run some commands.
For example,
#
python
Python 3.6.9 (default, Oct 8 2020, 12:12:24)
[GCC 8.4.0] on linux
Type “help”, “copyright”, “credits” or “license” for more information.
>>>
import pycuda
>>>
import pycuda.driver as cuda
>>>
cuda.init()
#
nvcc -V