Error setting up TAO Toolkit - 'nvidia-docker not found'

Hello,

I am setting up TAO Toolkit as per beginner guide. I am getting below docker errors and package missing errors.

For the package related error, I installed the package individually as below, which succeedes. But, the error is shown when doing the complete install.

I am sharing the complete log file from console.

INFO: Check requirements
INFO: Checking Python installation
INFO: python3 found.
INFO: Python version: Python 3.12.9
INFO: pip3 found.
INFO: Pip version: pip 25.0 from /home/abhilash/miniconda3/lib/python3.12/site-packages/pip (python 3.12)
INFO: Docker found. Checking additional requirements for docker.
INFO: Checking nvidia-docker2 installation
ERROR: nvidia-docker not found.
INFO: NGC CLI found.
INFO: NGC CLI 3.63.0
INFO: Requirements check satisfied. Installing TAO Toolkit.
By installing the TAO Toolkit CLI, you accept the terms and conditions of this license: https://developer.nvidia.com/tao-toolkit-software-license-agreement
Would you like to continue? (y/n): y
INFO: EULA accepted.
INFO: Installing TAO Toolkit CLI
INFO: jupyter installation was found.
Selected Jupyter core packages...
IPython          : 9.1.0
ipykernel        : not installed
ipywidgets       : 8.1.6
jupyter_client   : not installed
jupyter_core     : 5.7.2
jupyter_server   : 2.15.0
jupyterlab       : not installed
nbclient         : not installed
nbconvert        : 7.16.6
nbformat         : 5.10.4
notebook         : 7.4.0
qtconsole        : not installed
traitlets        : 5.14.3
INFO: TAO Toolkit was found
Traceback (most recent call last):
  File "/home/abhilash/miniconda3/bin/tao", line 5, in <module>
    from nvidia_tao_cli.entrypoint.tao_launcher import main
  File "/home/abhilash/miniconda3/lib/python3.12/site-packages/nvidia_tao_cli/entrypoint/tao_launcher.py", line 23, in <module>
    from nvidia_tao_cli.components.instance_handler.builder import get_launcher
  File "/home/abhilash/miniconda3/lib/python3.12/site-packages/nvidia_tao_cli/components/instance_handler/builder.py", line 24, in <module>
    from nvidia_tao_cli.components.instance_handler.local_instance import LocalInstance
  File "/home/abhilash/miniconda3/lib/python3.12/site-packages/nvidia_tao_cli/components/instance_handler/local_instance.py", line 29, in <module>
    from nvidia_tao_cli.components.docker_handler.docker_handler import (
  File "/home/abhilash/miniconda3/lib/python3.12/site-packages/nvidia_tao_cli/components/docker_handler/docker_handler.py", line 29, in <module>
    import docker
  File "/home/abhilash/miniconda3/lib/python3.12/site-packages/docker/__init__.py", line 2, in <module>
    from .api import APIClient
  File "/home/abhilash/miniconda3/lib/python3.12/site-packages/docker/api/__init__.py", line 2, in <module>
    from .client import APIClient
  File "/home/abhilash/miniconda3/lib/python3.12/site-packages/docker/api/client.py", line 8, in <module>
    import websocket
  File "/home/abhilash/miniconda3/lib/python3.12/site-packages/websocket/__init__.py", line 23, in <module>
    from ._app import WebSocketApp
  File "/home/abhilash/miniconda3/lib/python3.12/site-packages/websocket/_app.py", line 36, in <module>
    from ._core import WebSocket, getdefaulttimeout
  File "/home/abhilash/miniconda3/lib/python3.12/site-packages/websocket/_core.py", line 34, in <module>
    from ._handshake import *
  File "/home/abhilash/miniconda3/lib/python3.12/site-packages/websocket/_handshake.py", line 30, in <module>
    from ._http import *
  File "/home/abhilash/miniconda3/lib/python3.12/site-packages/websocket/_http.py", line 33, in <module>
    from ._url import *
  File "/home/abhilash/miniconda3/lib/python3.12/site-packages/websocket/_url.py", line 27, in <module>
    from six.moves.urllib.parse import urlparse
ModuleNotFoundError: No module named 'six.moves'
INFO:

Any thoughts on how to fix this?

Any help is appreciated. Thanks in advance.

1 post - 1 participant

Read full topic