Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Container exited instantly when running image without any error #3757

Closed
minhhotboy9x opened this issue Mar 31, 2024 · 1 comment
Closed

Comments

@minhhotboy9x
Copy link

Description

Hi.
I'm trying to run Tensorrt on Docker container on Win10. When I ran my image, it just showed an introduction and ended without any error. Can sb help me with it.
Here my dockerfile:

FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu20.04

ENV TZ=UTC

ARG DEBIAN_FRONTEND=noninteractive

RUN apt-get update -y && apt-get install -y software-properties-common

RUN add-apt-repository ppa:deadsnakes/ppa

RUN add-apt-repository ppa:deadsnakes/ppa && \
    apt-get update && \
    apt-get install -y python3.10 python3-pip

RUN pip install networkx==2.8.8

RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121

RUN pip install --upgrade pip

RUN pip install wheel tensorrt --index-url https://pypi.nvidia.com

WORKDIR /workspace

CMD [ "/bin/bash" ]

My cmd:
docker run --gpus all --ulimit stack=67108864 -v "D:/intern awl/week6/yolov8/":/workspace torch-tensorrt

My log:

==========
== CUDA ==
==========

CUDA Version 12.1.1

Container image Copyright (c) 2016-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.

This container image and its contents are governed by the NVIDIA Deep Learning Container License.
By pulling and using the container, you accept the terms and conditions of this license:
https://developer.nvidia.com/ngc/nvidia-deep-learning-container-license

A copy of this license is made available in this container at /NGC-DL-CONTAINER-LICENSE for your convenience.

Also, docker take too much RAM and Disk:
image

Environment

TensorRT Version:
8.6.1
NVIDIA GPU:
RTX 3050 laptop
NVIDIA Driver Version:

CUDA Version:
12.1.1
CUDNN Version:

Operating System:
Window 10
Python Version (if applicable):
3.10
Tensorflow Version (if applicable):

PyTorch Version (if applicable):

Baremetal or Container (if so, version):

Relevant Files

Model link:

Steps To Reproduce

Commands or scripts:

Have you tried the latest release?:

Can this model run on other frameworks? For example run ONNX model with ONNXRuntime (polygraphy run <model.onnx> --onnxrt):

@minhhotboy9x
Copy link
Author

I tried with another cmd:
docker run -it --gpus all --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 -v "D:/intern awl/week6/yolov8/":/workspace torch-tensorrt /bin/bash
and it worked

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant