TypeError: '>=' not supported between instances of 'str' and 'Version' (CUDA_VERSION not set)

[tokk-nv]

This is a known issue, so I'm filing this just for tracking purposes.

When you only install nvidia-container package and docker on a fresh Jetson Linux (L4T) setup, jetson-container will run into this error.

  File "/home/jetson/jetson-containers_tokknv/packages/cuda/cudnn/config.py", line 11, in <module>
    if CUDA_VERSION >= Version('12.6'):
TypeError: '>=' not supported between instances of 'str' and 'Version'

A workaround would be to set CUDA_VERSION in an environment variable, like

export CUDA_VERSION=12.4

but perhaps we may want to auto-set the expected CUDA_VERSION by default based on the L4T revision number.