NVIDIA / tao_tutorials

Quick start scripts and tutorial notebooks to get started with TAO Toolkit
Apache License 2.0
41 stars 7 forks source link

Installing Error #6

Open Doruk-Dilmen opened 5 months ago

Doruk-Dilmen commented 5 months ago

Hi,

I tried to install Tao Toolkit from this link: https://docs.nvidia.com/tao/tao-toolkit/text/tao_toolkit_quick_start_guide.html

When I tried this command, I get error: image

Error like this: image

I should use Tao Toolkit but if I couldn't, only Retail dataset and pretrained model can be enough .I just want to use retail pretrained model. I need this model because my school project depends this dataset. Can you help me? I need retail dataset and pretrained model. Can you share with me?

iamamiramine commented 1 day ago

First install nvidia-container-toolkit following the instructions from nvidia docs

Then update the setup/quickstart_launcher.sh script starting from info "Checking nvidia-docker2 installation" (line 114) change the following script:

    info "Checking nvidia-docker2 installation"
    if command -v nvidia-docker >/dev/null; then
        nvidia_docker_version=$(nvidia-docker version | grep "NVIDIA Docker:" | cut -d " " -f 3)
        [[ $nvidia_docker_version = 2* ]] || warning "nvidia-docker2 is required, current version is $nvidia_docker_version"
    else
        error "nvidia-docker not found."
    fi

to

    info "Checking nvidia-docker2 installation"
    if command -v docker >/dev/null; then
        if docker info --format '{{json .Runtimes}}' | grep -q nvidia; then
            echo "NVIDIA Container Toolkit is properly installed."
        else
            warning "NVIDIA Container Toolkit is not detected in Docker runtimes."
        fi

        # Check Docker version
        docker_version=$(docker version --format '{{.Server.Version}}')
        echo "Docker version: $docker_version"

        # Check NVIDIA Container Toolkit version
        nvidia_toolkit_version=$(dpkg -l | grep nvidia-container-toolkit | awk '{print $3}')
        if [ -n "$nvidia_toolkit_version" ]; then
            echo "NVIDIA Container Toolkit version: $nvidia_toolkit_version"
        else
            error "NVIDIA Container Toolkit not found."
        fi
    else
        error "Docker not found."
    fi