How to build custom image with specific python packages

[gianpaj]

Search before asking

• I have searched the jetson-containers issues and found no similar feature requests.

Question

For example

$ jetson-containers build --skip-tests=all pytorch conformer diffusers gradio librosa safetensors transformers

Namespace(packages=['pytorch', 'conformer', 'diffusers', 'gradio', 'librosa', 'safetensors', 'transformers'], name='', base='', multiple=False, build_flags='', build_args='', use_proxy=False, package_dirs=[''], list_packages=False, show_packages=False, skip_packages=[''], skip_errors=False, skip_tests=['all'], test_only=[''], simulate=False, push='', no_github_api=False, log_dir=None, log_level=None, log_colors=None, log_status=None, debug=False, verbose=False, version=False)

┌───────────────────────┬────────────────────────┐
│ L4T_VERSION   36.4.3  │ JETPACK_VERSION  6.2   │
│ CUDA_VERSION  12.6    │ PYTHON_VERSION   3.10  │
│ SYSTEM_ARCH   aarch64 │ LSB_RELEASE      22.04 │
└───────────────────────┴────────────────────────┘

$ jetson-containers --skip-tests=all pytorch conformer diffusers gradio librosa safetensors transformers

[17:39:45] Failed building: pytorch, conformer, diffusers, gradio, librosa, safetensors, transformers

Traceback (most recent call last):
  File "/ssd/jetson-containers/jetson_containers/build.py", line 129, in <module>
    build_container(**vars(args))
  File "/ssd/jetson-containers/jetson_containers/container.py", line 118, in build_container
    packages = resolve_dependencies(packages, skip_packages=skip_packages)
               ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/ssd/jetson-containers/jetson_containers/packages.py", line 392, in resolve_dependencies
    packages, changed = add_depends(packages)
                        ^^^^^^^^^^^^^^^^^^^^^
  File "/ssd/jetson-containers/jetson_containers/packages.py", line 363, in add_depends
    for dependency in find_package(package).get('depends', []):
                      ^^^^^^^^^^^^^^^^^^^^^
  File "/ssd/jetson-containers/jetson_containers/packages.py", line 247, in find_package
    raise KeyError(f"couldn't find package:  {package}")
KeyError: "couldn't find package:  conformer"

Additional

No response

[johnnynunez]

what is conformer?