andreyvelichkevich/test-torch:latest

Index digest

sha256:ff0c7908797790e492c76dc74fa8e56c9b8b2247949bbdb1ae40f24bbfe64b45

OS/ARCH

linux/amd64

Compressed size

3.53 GB

Last pushed

Type

Image

Manifest digest

sha256:975843f8f2a5ddfbdce2c519374dcf5a6306d49d12f80ccf54ae8423ad07cd10

Image Layers

1ARG RELEASE0 B

2ARG LAUNCHPAD_BUILD_ARCH0 B

3LABEL org.opencontainers.image.ref.name=ubuntu0 B

4LABEL org.opencontainers.image.version=20.040 B

5ADD file ... in / 27.26 MB

6CMD ["/bin/bash"]0 B

7ARG PYTORCH_VERSION0 B

8ARG TRITON_VERSION0 B

9ARG TARGETPLATFORM0 B

10ARG CUDA_VERSION0 B

11LABEL com.nvidia.volumes.needed=nvidia_driver0 B

12RUN |4 PYTORCH_VERSION=2.1.2 TRITON_VERSION=2.1.0+e6216047b8 TARGETPLATFORM=linux/amd649.55 MB

13COPY /opt/conda /opt/conda # buildkit3.49 GB

14RUN |4 PYTORCH_VERSION=2.1.2 TRITON_VERSION=2.1.0+e6216047b8 TARGETPLATFORM=linux/amd6432 B

15ENV PATH=/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B

16ENV NVIDIA_VISIBLE_DEVICES=all0 B

17ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility0 B

18ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib640 B

19ENV PYTORCH_VERSION=2.1.20 B

20WORKDIR /workspace99 B

21WORKDIR /app93 B

22ADD train.py train.py # buildkit421 B

23ENTRYPOINT ["torchrun" "train.py"]0 B

Command

 ARG RELEASE