docker.io/nvcr.io/nvidia/pytorch:24.06-py3 linux/amd64

docker.io/nvcr.io/nvidia/pytorch:24.06-py3 - 国内下载镜像源 浏览次数:80
这里是镜像的描述信息： NVIDIA PyTorch Docker Image

这是一个基于PyTorch框架的Docker容器镜像，提供了一个完整的深度学习环境。该镜像包含了PyTorch 1.x版本，以及其他必需的依赖包，如CUDA、cuDNN等。

使用这个镜像，您可以轻松地在本地环境中搭建一个深度学习工作站，进行各种机器学习和计算机视觉任务的实验和开发。

此外，该镜像还支持GPU加速，通过NVIDIA的CUDA和cuDNN技术，可以显著提高PyTorch的性能和效率。

源镜像 docker.io/nvcr.io/nvidia/pytorch:24.06-py3
国内镜像 swr.cn-north-4.myhuaweicloud.com/ddn-k8s/docker.io/nvcr.io/nvidia/pytorch:24.06-py3
镜像ID sha256:7598d7ac994bf5ac43a5550c2ea4d35a7bb0aad419288d82ff2401a83752620d
镜像TAG 24.06-py3
大小 19.15GB
镜像源 docker.io
项目信息 Docker-Hub主页 🚀项目TAG 🚀
CMD
启动入口 /opt/nvidia/nvidia_entrypoint.sh
工作目录 /workspace
OS/平台 linux/amd64
浏览量 80 次
贡献者
镜像创建 2024-06-07T01:17:53.042480121Z
同步时间 2024-10-22 00:29
更新时间 2024-10-23 19:10
开放端口
6006/tcp 8888/tcp
环境变量
PATH=/usr/local/lib/python3.10/dist-packages/torch_tensorrt/bin:/usr/local/mpi/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin:/opt/tensorrt/bin CUDA_VERSION=12.5.0.023 CUDA_DRIVER_VERSION=555.42.02 CUDA_CACHE_DISABLE=1 NVIDIA_REQUIRE_JETPACK_HOST_MOUNTS= _CUDA_COMPAT_PATH=/usr/local/cuda/compat ENV=/etc/shinit_v2 BASH_ENV=/etc/bash.bashrc SHELL=/bin/bash NVIDIA_REQUIRE_CUDA=cuda>=9.0 NCCL_VERSION=2.21.5 CUBLAS_VERSION=12.5.2.13 CUFFT_VERSION=11.2.3.18 CURAND_VERSION=10.3.6.39 CUSPARSE_VERSION=12.4.1.24 CUSOLVER_VERSION=11.6.2.40 CUTENSOR_VERSION=2.0.1.2 NPP_VERSION=12.3.0.116 NVJPEG_VERSION=12.3.2.38 CUDNN_VERSION=9.1.0.70+cuda12.4 TRT_VERSION=10.1.0.27+cuda12.4.1.003 TRTOSS_VERSION=24.06 NSIGHT_SYSTEMS_VERSION=2024.2.3.38 NSIGHT_COMPUTE_VERSION=2024.2.0.16 DALI_VERSION=1.38.0 DALI_BUILD=15028468 POLYGRAPHY_VERSION=0.49.11 TRANSFORMER_ENGINE_VERSION=1.7 MODEL_OPT_VERSION=0.11.2 LD_LIBRARY_PATH=/usr/local/lib/python3.10/dist-packages/torch/lib:/usr/local/lib/python3.10/dist-packages/torch_tensorrt/lib:/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64 NVIDIA_VISIBLE_DEVICES=all NVIDIA_DRIVER_CAPABILITIES=compute,utility,video NVIDIA_PRODUCT_NAME=PyTorch GDRCOPY_VERSION=2.3.1-1 HPCX_VERSION=2.19 MOFED_VERSION=5.4-rdmacore39.0 OPENUCX_VERSION=1.17.0 OPENMPI_VERSION=4.1.7 RDMACORE_VERSION=39.0 OPAL_PREFIX=/opt/hpcx/ompi OMPI_MCA_coll_hcoll_enable=0 LIBRARY_PATH=/usr/local/cuda/lib64/stubs: PYTORCH_BUILD_VERSION=2.4.0a0+f70bd71 PYTORCH_VERSION=2.4.0a0+f70bd71 PYTORCH_BUILD_NUMBER=0 NVIDIA_PYTORCH_VERSION=24.06 NVFUSER_BUILD_VERSION=f73ff1b NVFUSER_VERSION=f73ff1b PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python NVPL_LAPACK_MATH_MODE=PEDANTIC PYTHONIOENCODING=utf-8 LC_ALL=C.UTF-8 PIP_DEFAULT_TIMEOUT=100 NVM_DIR=/usr/local/nvm NODE_OPTIONS=--openssl-legacy-provider JUPYTER_PORT=8888 TENSORBOARD_PORT=6006 UCC_CL_BASIC_TLS=^sharp TORCH_CUDA_ARCH_LIST=5.2 6.0 6.1 7.0 7.2 7.5 8.0 8.6 8.7 9.0+PTX PYTORCH_HOME=/opt/pytorch/pytorch CUDA_HOME=/usr/local/cuda TORCH_ALLOW_TF32_CUBLAS_OVERRIDE=1 USE_EXPERIMENTAL_CUDNN_V8_API=1 COCOAPI_VERSION=2.0+nv0.8.0 TORCH_CUDNN_V8_API_ENABLED=1 CUDA_MODULE_LOADING=LAZY NVIDIA_BUILD_ID=96418707
镜像标签
96418707: com.nvidia.build.id 3017efeca4038be2c8c0f965cf868f48051bff16: com.nvidia.build.ref 12.5.2.13: com.nvidia.cublas.version 9.0: com.nvidia.cuda.version 9.1.0.70+cuda12.4: com.nvidia.cudnn.version 11.2.3.18: com.nvidia.cufft.version 10.3.6.39: com.nvidia.curand.version 11.6.2.40: com.nvidia.cusolver.version 12.4.1.24: com.nvidia.cusparse.version 2.0.1.2: com.nvidia.cutensor.version 2.21.5: com.nvidia.nccl.version 12.3.0.116: com.nvidia.npp.version 2024.2.0.16: com.nvidia.nsightcompute.version 2024.2.3.38: com.nvidia.nsightsystems.version 12.3.2.38: com.nvidia.nvjpeg.version 2.4.0a0+f70bd71: com.nvidia.pytorch.version 10.1.0.27+cuda12.4.1.003: com.nvidia.tensorrt.version 24.06: com.nvidia.tensorrtoss.version nvidia_driver: com.nvidia.volumes.needed ubuntu: org.opencontainers.image.ref.name 22.04: org.opencontainers.image.version

Docker拉取命令 无权限下载?点我修复 

docker pull swr.cn-north-4.myhuaweicloud.com/ddn-k8s/docker.io/nvcr.io/nvidia/pytorch:24.06-py3
docker tag  swr.cn-north-4.myhuaweicloud.com/ddn-k8s/docker.io/nvcr.io/nvidia/pytorch:24.06-py3  docker.io/nvcr.io/nvidia/pytorch:24.06-py3

Containerd拉取命令 

ctr images pull swr.cn-north-4.myhuaweicloud.com/ddn-k8s/docker.io/nvcr.io/nvidia/pytorch:24.06-py3
ctr images tag  swr.cn-north-4.myhuaweicloud.com/ddn-k8s/docker.io/nvcr.io/nvidia/pytorch:24.06-py3  docker.io/nvcr.io/nvidia/pytorch:24.06-py3

Shell快速替换命令 

sed -i 's#nvcr.io/nvidia/pytorch:24.06-py3#swr.cn-north-4.myhuaweicloud.com/ddn-k8s/docker.io/nvcr.io/nvidia/pytorch:24.06-py3#' deployment.yaml

镜像历史

大小 创建时间 层信息
0.00B 2024-06-07 09:17:53 LABEL com.nvidia.build.ref=3017efeca4038be2c8c0f965cf868f48051bff16
0.00B 2024-06-07 09:17:53 ARG NVIDIA_BUILD_REF
0.00B 2024-06-07 09:17:53 LABEL com.nvidia.build.id=96418707
0.00B 2024-06-07 09:17:53 ENV NVIDIA_BUILD_ID=96418707
0.00B 2024-06-07 09:17:53 ARG NVIDIA_BUILD_ID
719.00B 2024-06-07 09:17:53 COPY entrypoint.d/ /opt/nvidia/entrypoint.d/ # buildkit
83.90KB 2024-06-07 09:17:52 RUN |1 PYVER=3.10 /bin/sh -c ln -sf ${_CUDA_COMPAT_PATH}/lib.real ${_CUDA_COMPAT_PATH}/lib && echo ${_CUDA_COMPAT_PATH}/lib > /etc/ld.so.conf.d/00-cuda-compat.conf && ldconfig && rm -f ${_CUDA_COMPAT_PATH}/lib # buildkit
0.00B 2024-06-07 09:17:52 ENV CUDA_MODULE_LOADING=LAZY
0.00B 2024-06-07 09:17:52 ENV TORCH_CUDNN_V8_API_ENABLED=1
342.36MB 2024-06-07 09:17:52 RUN |1 PYVER=3.10 /bin/sh -c if [ "${L4T}" = "1" ]; then echo "Not installing Transformer Engine in iGPU container until Version variable is set"; else pip install --no-cache-dir --no-build-isolation git+https://github.com/NVIDIA/TransformerEngine.git@release_v${TRANSFORMER_ENGINE_VERSION}; fi # buildkit
399.83MB 2024-06-07 09:10:53 RUN |1 PYVER=3.10 /bin/sh -c if [ "${L4T}" = "1" ]; then echo "Not installing Flash Attention in iGPU as it is a requirement for Transformer Engine"; else env MAX_JOBS=4 pip install flash-attn==2.4.2; fi # buildkit
0.00B 2024-06-07 08:22:20 ENV PATH=/usr/local/lib/python3.10/dist-packages/torch_tensorrt/bin:/usr/local/mpi/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin:/opt/tensorrt/bin
0.00B 2024-06-07 08:22:20 ENV LD_LIBRARY_PATH=/usr/local/lib/python3.10/dist-packages/torch/lib:/usr/local/lib/python3.10/dist-packages/torch_tensorrt/lib:/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64
44.99MB 2024-06-07 08:22:20 RUN |1 PYVER=3.10 /bin/sh -c pip install --no-cache-dir /opt/pytorch/torch_tensorrt/dist/*.whl # buildkit
0.00B 2024-06-07 08:17:59 ARG PYVER
150.03MB 2024-06-07 08:17:59 COPY torch_tensorrt/ /opt/pytorch/torch_tensorrt/ # buildkit
20.30MB 2024-06-07 08:17:58 RUN /bin/sh -c pip --version && python -c 'import sys; print(sys.platform)' && pip install --no-cache-dir nvidia-pyindex && if [ "${L4T}" = "1" ]; then pip install polygraphy; else pip install --extra-index-url https://urm.nvidia.com/artifactory/api/pypi/sw-tensorrt-pypi/simple --no-cache-dir polygraphy==${POLYGRAPHY_VERSION}; fi && pip install --extra-index-url https://pypi.nvidia.com "nvidia-modelopt[torch]==${MODEL_OPT_VERSION}" && pip install --extra-index-url http://sqrl/dldata/pip-simple --trusted-host sqrl --no-cache-dir pytorch-quantization==2.1.2 # buildkit
0.00B 2024-06-07 08:17:35 ENV PATH=/usr/local/mpi/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin:/opt/tensorrt/bin
5.45MB 2024-06-07 08:17:35 RUN /bin/sh -c set -x && URL=$(VERIFY=1 /nvidia/build-scripts/installTRT.sh | sed -n "s/^.*\(http.*\)tar.*$/\1/p")tar && FILE=$(wget -O - $URL | sed -n 's/^.*href="\(TensorRT[^"]*\)".*$/\1/p' | egrep -v "internal|safety") && wget -q $URL/$FILE -O - | tar -xz && PY=$(python -c 'import sys; print(str(sys.version_info[0])+str(sys.version_info[1]))') && pip install TensorRT-*/python/tensorrt-*-cp$PY*.whl && mv /usr/src/tensorrt /opt && ln -s /opt/tensorrt /usr/src/tensorrt && rm -r TensorRT-* # buildkit
51.00MB 2024-06-07 08:16:49 RUN /bin/sh -c chmod -R a+w . # buildkit
34.89MB 2024-06-07 08:16:49 COPY tutorials tutorials # buildkit
15.96MB 2024-06-07 08:16:49 COPY examples examples # buildkit
2.07KB 2024-06-07 08:16:49 COPY docker-examples docker-examples # buildkit
2.05KB 2024-06-07 08:16:48 COPY NVREADME.md README.md # buildkit
0.00B 2024-06-07 08:16:48 WORKDIR /workspace
2.22GB 2024-06-07 08:16:48 RUN /bin/sh -c if [ "${L4T}" = "1" ]; then echo "Not installing rapids for L4T build." ; else find /rapids -name "*-Linux.tar.gz" -exec tar -C /usr --exclude="*.a" --exclude="bin/xgboost" --strip-components=1 -xvf {} \; && find /rapids -name "*.whl" ! -name "Pillow-*" ! -name "certifi-*" ! -name "protobuf-*" -exec pip install --no-cache-dir {} + ; pip install numpy==1.24.4; fi # buildkit
201.84KB 2024-06-07 08:16:08 RUN /bin/sh -c pip install --no-cache-dir --disable-pip-version-check tabulate # buildkit
747.16MB 2024-06-07 08:16:05 RUN /bin/sh -c ( cd vision && export PYTORCH_VERSION=$(python -c "import torch; print(torch.__version__)") && CFLAGS="-g0" FORCE_CUDA=1 NVCC_APPEND_FLAGS="--threads 8" pip install --no-cache-dir --no-build-isolation --disable-pip-version-check . ) && ( cd vision && cmake -Bbuild -H. -GNinja -DWITH_CUDA=1 -DCMAKE_PREFIX_PATH=`python -c 'import torch;print(torch.utils.cmake_prefix_path)'` && cmake --build build --target install && rm -rf build ) && ( cd fuser && pip install -r requirements.txt && python setup.py -version-tag=a0+${NVFUSER_VERSION} install && python setup.py clean && cp $(find /usr/local/lib/python3.10/dist-packages/ -name libnvfuser_codegen.so) /usr/local/lib/python3.10/dist-packages/torch/lib/ ) && ( cd lightning-thunder && python setup.py install && rm -rf build) && BUILD_OPTIONS="--cpp_ext --cuda_ext --bnp --xentropy --deprecated_fused_adam --deprecated_fused_lamb --fast_multihead_attn --distributed_lamb --fast_layer_norm --transducer --distributed_adam --fmha --permutation_search --focal_loss --fused_conv_bias_relu --index_mul_2d --cudnn_gbn --group_norm --gpu_direct_storage" && if [ "${L4T}" != "1" ]; then BUILD_OPTIONS="--fast_bottleneck --nccl_p2p --peer_memory --nccl_allocator ${BUILD_OPTIONS}"; fi && ( cd apex && CFLAGS="-g0" NVCC_APPEND_FLAGS="--threads 8" pip install -v --no-build-isolation --no-cache-dir --disable-pip-version-check --config-settings "--build-option=${BUILD_OPTIONS}" . && rm -rf build ) && ( cd lightning-thunder && mkdir tmp && cd tmp && git clone -b v1.3.0 --recursive --single-branch https://github.com/NVIDIA/cudnn-frontend.git cudnn_frontend && cd cudnn_frontend && pip install --no-build-isolation --no-cache-dir --disable-pip-version-check . && cd ../../ && rm -rf tmp ) && ( cd pytorch/third_party/onnx && pip uninstall typing -y && CMAKE_ARGS="-DONNX_USE_PROTOBUF_SHARED_LIBS=ON" pip install --no-build-isolation --no-cache-dir --disable-pip-version-check . ) # buildkit
2.21KB 2024-06-07 07:29:52 COPY singularity/ /.singularity.d/ # buildkit
79.06MB 2024-06-07 07:29:52 RUN /bin/sh -c export COCOAPI_TAG=$(echo ${COCOAPI_VERSION} | sed 's/^.*+n//') && pip install --disable-pip-version-check --no-cache-dir git+https://github.com/nvidia/cocoapi.git@${COCOAPI_TAG}#subdirectory=PythonAPI # buildkit
0.00B 2024-06-07 07:29:24 ENV COCOAPI_VERSION=2.0+nv0.8.0
652.61MB 2024-06-07 07:29:24 RUN /bin/sh -c if [ -z "${DALI_VERSION}" ] ; then echo "Not Installing DALI for L4T Build." ; else export DALI_PKG_SUFFIX="cuda${CUDA_VERSION%%.*}0" && pip install --disable-pip-version-check --no-cache-dir --extra-index-url https://developer.download.nvidia.com/compute/redist --extra-index-url http://sqrl/dldata/pip-dali${DALI_URL_SUFFIX:-} --trusted-host sqrl nvidia-dali-${DALI_PKG_SUFFIX}==${DALI_VERSION}; fi # buildkit
563.46MB 2024-06-07 07:29:12 RUN /bin/sh -c pip install --no-cache-dir /tmp/dist/*.whl # buildkit
122.90MB 2024-06-07 07:29:04 RUN |6 NVIDIA_PYTORCH_VERSION=24.06 PYTORCH_BUILD_VERSION=2.4.0a0+f70bd71 NVFUSER_BUILD_VERSION=f73ff1b TARGETARCH=amd64 PYVER=3.10 L4T=0 /bin/sh -c cd pytorch && bash .ci/docker/common/install_cusparselt.sh && pip install --no-cache-dir -v -r /opt/pytorch/pytorch/requirements.txt # buildkit
2.25GB 2024-06-07 07:28:58 RUN |6 NVIDIA_PYTORCH_VERSION=24.06 PYTORCH_BUILD_VERSION=2.4.0a0+f70bd71 NVFUSER_BUILD_VERSION=f73ff1b TARGETARCH=amd64 PYVER=3.10 L4T=0 /bin/sh -c pip install /opt/transfer/torch*.whl && patchelf --set-rpath '/usr/local/lib' /usr/local/lib/python3.10/dist-packages/torch/lib/libtorch_global_deps.so # buildkit
0.00B 2024-06-07 07:28:24 ENV USE_EXPERIMENTAL_CUDNN_V8_API=1
0.00B 2024-06-07 07:28:24 ENV TORCH_ALLOW_TF32_CUBLAS_OVERRIDE=1
0.00B 2024-06-07 07:28:24 ENV CUDA_HOME=/usr/local/cuda
0.00B 2024-06-07 07:28:24 ENV PYTORCH_HOME=/opt/pytorch/pytorch
0.00B 2024-06-07 07:28:24 ENV TORCH_CUDA_ARCH_LIST=5.2 6.0 6.1 7.0 7.2 7.5 8.0 8.6 8.7 9.0+PTX
0.00B 2024-06-07 07:28:24 ENV UCC_CL_BASIC_TLS=^sharp
53.68MB 2024-06-07 07:28:24 RUN |6 NVIDIA_PYTORCH_VERSION=24.06 PYTORCH_BUILD_VERSION=2.4.0a0+f70bd71 NVFUSER_BUILD_VERSION=f73ff1b TARGETARCH=amd64 PYVER=3.10 L4T=0 /bin/sh -c OPENCV_VERSION=4.7.0 && cd / && wget -q -O - https://github.com/opencv/opencv/archive/${OPENCV_VERSION}.tar.gz | tar -xzf - && cd /opencv-${OPENCV_VERSION} && cmake -GNinja -Bbuild -H. -DWITH_CUDA=OFF -DWITH_1394=OFF -DPYTHON3_PACKAGES_PATH="/usr/local/lib/python${PYVER}/dist-packages" -DBUILD_opencv_cudalegacy=OFF -DBUILD_opencv_stitching=OFF -DWITH_IPP=OFF -DWITH_PROTOBUF=OFF && cmake --build build --target install && cd modules/python/package && pip install --no-cache-dir --disable-pip-version-check -v . && rm -rf /opencv-${OPENCV_VERSION} # buildkit
0.00B 2024-06-07 07:24:36 EXPOSE map[6006/tcp:{}]
0.00B 2024-06-07 07:24:36 EXPOSE map[8888/tcp:{}]
0.00B 2024-06-07 07:24:36 ENV TENSORBOARD_PORT=6006
0.00B 2024-06-07 07:24:36 ENV JUPYTER_PORT=8888
427.00B 2024-06-07 07:24:36 COPY jupyter_notebook_config.py /usr/local/etc/jupyter/ # buildkit
217.62MB 2024-06-07 07:24:36 RUN |6 NVIDIA_PYTORCH_VERSION=24.06 PYTORCH_BUILD_VERSION=2.4.0a0+f70bd71 NVFUSER_BUILD_VERSION=f73ff1b TARGETARCH=amd64 PYVER=3.10 L4T=0 /bin/sh -c pip install --disable-pip-version-check --no-cache-dir git+https://github.com/cliffwoolley/jupyter_tensorboard.git@0.2.0+nv21.03 && mkdir -p $NVM_DIR && curl -Lo- https://raw.githubusercontent.com/nvm-sh/nvm/v0.39.7/install.sh | bash && source "$NVM_DIR/nvm.sh" && nvm install lts/hydrogen node && jupyter labextension install jupyterlab_tensorboard && jupyter serverextension enable jupyterlab && pip install --no-cache-dir jupytext && jupyter labextension install jupyterlab-jupytext@1.2.2 && ( cd /usr/local/share/jupyter/lab/staging && npm prune --production ) && npm cache clean --force && rm -rf /usr/local/share/.cache && echo "source $NVM_DIR/nvm.sh" >> /etc/bash.bashrc && mv /root/.jupyter/jupyter_notebook_config.json /usr/local/etc/jupyter/ && jupyter lab clean # buildkit
0.00B 2024-06-04 14:14:16 ENV NODE_OPTIONS=--openssl-legacy-provider
0.00B 2024-06-04 14:14:16 ENV NVM_DIR=/usr/local/nvm
27.51KB 2024-06-04 14:14:16 RUN |6 NVIDIA_PYTORCH_VERSION=24.06 PYTORCH_BUILD_VERSION=2.4.0a0+f70bd71 NVFUSER_BUILD_VERSION=f73ff1b TARGETARCH=amd64 PYVER=3.10 L4T=0 /bin/sh -c PATCHED_FILE=$(python -c "from tensorboard.plugins.core import core_plugin as _; print(_.__file__)") && sed -i 's/^\( *"--bind_all",\)$/\1 default=True,/' "$PATCHED_FILE" && test $(grep '^ *"--bind_all", default=True,$' "$PATCHED_FILE" | wc -l) -eq 1 # buildkit
175.89MB 2024-06-04 14:14:15 RUN |6 NVIDIA_PYTORCH_VERSION=24.06 PYTORCH_BUILD_VERSION=2.4.0a0+f70bd71 NVFUSER_BUILD_VERSION=f73ff1b TARGETARCH=amd64 PYVER=3.10 L4T=0 /bin/sh -c git config --global url."https://github".insteadOf git://github && pip install --no-cache-dir notebook==6.4.10 jupyterlab==2.3.2 python-hostlist traitlets==5.9.0 && pip install --no-cache-dir tensorboard==2.9.0 # buildkit
2.15GB 2024-06-04 14:13:58 RUN |6 NVIDIA_PYTORCH_VERSION=24.06 PYTORCH_BUILD_VERSION=2.4.0a0+f70bd71 NVFUSER_BUILD_VERSION=f73ff1b TARGETARCH=amd64 PYVER=3.10 L4T=0 /bin/sh -c pip install --no-cache-dir numpy==1.24.4 scipy==1.11.3 "PyYAML>=5.4.1" astunparse typing_extensions cffi spacy mock tqdm librosa==0.10.1 expecttest==0.1.3 hypothesis==5.35.1 xdoctest==1.0.2 pytest==8.1.1 pytest-xdist pytest-rerunfailures pytest-shard pytest-flakefinder pybind11 Cython "regex>=2020.1.8" protobuf==4.24.4 && if [[ $TARGETARCH = "amd64" ]] ; then pip install --no-cache-dir mkl==2021.1.1 mkl-include==2021.1.1 mkl-devel==2021.1.1 ; find /usr/local/lib -maxdepth 1 -type f -regex '.*\/lib\(tbb\|mkl\).*\.so\($\|\.[0-9]*\.[0-9]*\)' -exec rm -v {} + ; fi # buildkit
0.00B 2024-06-04 14:13:10 ENV PIP_DEFAULT_TIMEOUT=100
0.00B 2024-06-04 14:13:10 ENV LC_ALL=C.UTF-8
0.00B 2024-06-04 14:13:10 ENV PYTHONIOENCODING=utf-8
1.43GB 2024-06-04 14:13:10 COPY . . # buildkit
0.00B 2024-06-04 14:13:02 WORKDIR /opt/pytorch
0.00B 2024-06-04 14:56:00 ENV NVPL_LAPACK_MATH_MODE=PEDANTIC
0.00B 2024-06-04 14:56:00 RUN |6 NVIDIA_PYTORCH_VERSION=24.06 PYTORCH_BUILD_VERSION=2.4.0a0+f70bd71 NVFUSER_BUILD_VERSION=f73ff1b TARGETARCH=amd64 PYVER=3.10 L4T=0 /bin/sh -c if [ $TARGETARCH = "arm64" ]; then cd /opt && curl "https://gitlab-master.nvidia.com/api/v4/projects/105799/packages/generic/nvpl_slim_24.04/sbsa/nvpl_slim_24.04.tar" --output nvpl_slim_24.04.tar && tar -xf nvpl_slim_24.04.tar && cp -r nvpl_slim_24.04/lib/* /usr/local/lib && cp -r nvpl_slim_24.04/include/* /usr/local/include && rm -rf nvpl_slim_24.04.tar nvpl_slim_24.04 ; fi # buildkit
46.71MB 2024-06-04 14:13:02 RUN |6 NVIDIA_PYTORCH_VERSION=24.06 PYTORCH_BUILD_VERSION=2.4.0a0+f70bd71 NVFUSER_BUILD_VERSION=f73ff1b TARGETARCH=amd64 PYVER=3.10 L4T=0 /bin/sh -c curl "https://gitlab-master.nvidia.com/api/v4/projects/105799/packages/generic/OpenBLAS/0.3.24-$(uname -m)/OpenBLAS-0.3.24-$(uname -m).tar.gz" --output OpenBLAS.tar.gz && tar -xf OpenBLAS.tar.gz -C /usr/local/ && rm OpenBLAS.tar.gz # buildkit
70.35MB 2024-06-04 14:13:01 RUN |6 NVIDIA_PYTORCH_VERSION=24.06 PYTORCH_BUILD_VERSION=2.4.0a0+f70bd71 NVFUSER_BUILD_VERSION=f73ff1b TARGETARCH=amd64 PYVER=3.10 L4T=0 /bin/sh -c pip install --no-cache-dir pip setuptools==68.2.2 && pip install --no-cache-dir cmake # buildkit
21.03MB 2024-06-04 14:12:56 RUN |6 NVIDIA_PYTORCH_VERSION=24.06 PYTORCH_BUILD_VERSION=2.4.0a0+f70bd71 NVFUSER_BUILD_VERSION=f73ff1b TARGETARCH=amd64 PYVER=3.10 L4T=0 /bin/sh -c curl -O https://bootstrap.pypa.io/get-pip.py && python get-pip.py && rm get-pip.py # buildkit
0.00B 2024-06-04 14:12:52 ENV PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python
198.72MB 2024-06-04 14:12:52 RUN |6 NVIDIA_PYTORCH_VERSION=24.06 PYTORCH_BUILD_VERSION=2.4.0a0+f70bd71 NVFUSER_BUILD_VERSION=f73ff1b TARGETARCH=amd64 PYVER=3.10 L4T=0 /bin/sh -c export PYSFX=`echo "$PYVER" | cut -c1-1` && export DEBIAN_FRONTEND=noninteractive && apt-get update && apt-get install -y --no-install-recommends python$PYVER-dev python$PYSFX python$PYSFX-dev python$PYSFX-distutils python-is-python$PYSFX autoconf automake libatlas-base-dev libgoogle-glog-dev libbz2-dev libc-ares2 libre2-9 libleveldb-dev liblmdb-dev libprotobuf-dev libsnappy-dev libtool nasm protobuf-compiler pkg-config unzip sox libsndfile1 libpng-dev libhdf5-103 libhdf5-dev gfortran rapidjson-dev ninja-build libedit-dev build-essential patchelf && rm -rf /var/lib/apt/lists/* # buildkit
0.00B 2024-06-04 14:12:52 ARG L4T=0
0.00B 2024-06-04 14:12:52 ARG PYVER=3.10
0.00B 2024-06-04 14:12:52 ARG TARGETARCH
0.00B 2024-06-04 14:12:52 LABEL com.nvidia.pytorch.version=2.4.0a0+f70bd71
0.00B 2024-06-04 14:12:52 ENV NVFUSER_BUILD_VERSION=f73ff1b NVFUSER_VERSION=f73ff1b
0.00B 2024-06-04 14:12:52 ENV PYTORCH_BUILD_VERSION=2.4.0a0+f70bd71 PYTORCH_VERSION=2.4.0a0+f70bd71 PYTORCH_BUILD_NUMBER=0 NVIDIA_PYTORCH_VERSION=24.06
0.00B 2024-06-04 14:12:52 ARG NVFUSER_BUILD_VERSION
0.00B 2024-06-04 14:12:52 ARG PYTORCH_BUILD_VERSION
0.00B 2024-06-04 14:12:52 ARG NVIDIA_PYTORCH_VERSION
0.00B 2024-06-04 14:12:52 ENV NVIDIA_PRODUCT_NAME=PyTorch
0.00B 2024-06-04 01:29:19 ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs:
995.54MB 2024-06-04 01:29:19 RUN |7 GDRCOPY_VERSION=2.3.1-1 HPCX_VERSION=2.19 RDMACORE_VERSION=39.0 MOFED_VERSION=5.4-rdmacore39.0 OPENUCX_VERSION=1.17.0 OPENMPI_VERSION=4.1.7 TARGETARCH=amd64 /bin/sh -c export DEVEL=1 BASE=0 && /nvidia/build-scripts/installNCU.sh && /nvidia/build-scripts/installCUDA.sh && /nvidia/build-scripts/installLIBS.sh && if [ ! -f /etc/ld.so.conf.d/nvidia-tegra.conf ]; then /nvidia/build-scripts/installNCCL.sh; fi && /nvidia/build-scripts/installCUDNN.sh && /nvidia/build-scripts/installCUTENSOR.sh && /nvidia/build-scripts/installTRT.sh && /nvidia/build-scripts/installNSYS.sh && if [ -f "/tmp/cuda-${_CUDA_VERSION_MAJMIN}.patch" ]; then patch -p0 < /tmp/cuda-${_CUDA_VERSION_MAJMIN}.patch; fi && rm -f /tmp/cuda-*.patch # buildkit
1.49KB 2024-06-04 01:24:47 COPY cuda-*.patch /tmp # buildkit
0.00B 2024-06-04 01:24:47 ENV OMPI_MCA_coll_hcoll_enable=0
0.00B 2024-06-04 01:24:47 ENV OPAL_PREFIX=/opt/hpcx/ompi PATH=/usr/local/mpi/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin
229.15MB 2024-06-04 01:24:47 RUN |7 GDRCOPY_VERSION=2.3.1-1 HPCX_VERSION=2.19 RDMACORE_VERSION=39.0 MOFED_VERSION=5.4-rdmacore39.0 OPENUCX_VERSION=1.17.0 OPENMPI_VERSION=4.1.7 TARGETARCH=amd64 /bin/sh -c cd /nvidia && ( export DEBIAN_FRONTEND=noninteractive && apt-get update && apt-get install -y --no-install-recommends libibverbs1 libibverbs-dev librdmacm1 librdmacm-dev libibumad3 libibumad-dev ibverbs-utils ibverbs-providers && rm -rf /var/lib/apt/lists/* && rm $(dpkg-query -L libibverbs-dev librdmacm-dev libibumad-dev | grep "\(\.so\|\.a\)$") ) && ( cd opt/gdrcopy/ && dpkg -i libgdrapi_*.deb ) && ( cp -r opt/hpcx /opt/ && cp etc/ld.so.conf.d/hpcx.conf /etc/ld.so.conf.d/ && ln -sf /opt/hpcx/ompi /usr/local/mpi && ln -sf /opt/hpcx/ucx /usr/local/ucx && sed -i 's/^\(hwloc_base_binding_policy\) = core$/\1 = none/' /opt/hpcx/ompi/etc/openmpi-mca-params.conf && sed -i 's/^\(btl = self\)$/#\1/' /opt/hpcx/ompi/etc/openmpi-mca-params.conf ) && ldconfig # buildkit
0.00B 2024-06-04 01:24:47 ARG TARGETARCH
0.00B 2024-06-04 01:24:47 ENV GDRCOPY_VERSION=2.3.1-1 HPCX_VERSION=2.19 MOFED_VERSION=5.4-rdmacore39.0 OPENUCX_VERSION=1.17.0 OPENMPI_VERSION=4.1.7 RDMACORE_VERSION=39.0
0.00B 2024-06-04 01:24:47 ARG OPENMPI_VERSION
0.00B 2024-06-04 01:24:47 ARG OPENUCX_VERSION
0.00B 2024-06-04 01:24:47 ARG MOFED_VERSION=5.4-rdmacore39.0
0.00B 2024-06-04 01:24:47 ARG RDMACORE_VERSION
0.00B 2024-06-04 01:24:47 ARG HPCX_VERSION
0.00B 2024-06-04 01:24:47 ARG GDRCOPY_VERSION
84.89MB 2024-06-04 01:22:44 RUN /bin/sh -c export DEBIAN_FRONTEND=noninteractive && apt-get update && apt-get install -y --no-install-recommends build-essential git libglib2.0-0 less libnl-route-3-200 libnl-3-dev libnl-route-3-dev libnuma-dev libnuma1 libpmi2-0-dev nano numactl openssh-client vim wget && rm -rf /var/lib/apt/lists/* # buildkit
148.72KB 2024-06-04 01:08:59 COPY NVIDIA_Deep_Learning_Container_License.pdf /workspace/ # buildkit
0.00B 2024-06-04 01:08:59 ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]
0.00B 2024-06-04 01:08:59 ENV NVIDIA_PRODUCT_NAME=CUDA
14.85KB 2024-06-04 01:08:59 COPY entrypoint/ /opt/nvidia/ # buildkit
0.00B 2024-06-04 01:08:59 ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin LD_LIBRARY_PATH=/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64 NVIDIA_VISIBLE_DEVICES=all NVIDIA_DRIVER_CAPABILITIES=compute,utility,video
0.00B 2024-06-04 01:08:59 ARG _LIBPATH_SUFFIX
46.00B 2024-06-04 01:08:59 RUN |22 CUDA_VERSION=12.5.0.023 CUDA_DRIVER_VERSION=555.42.02 JETPACK_HOST_MOUNTS= NCCL_VERSION=2.21.5 CUBLAS_VERSION=12.5.2.13 CUFFT_VERSION=11.2.3.18 CURAND_VERSION=10.3.6.39 CUSPARSE_VERSION=12.4.1.24 CUSOLVER_VERSION=11.6.2.40 CUTENSOR_VERSION=2.0.1.2 NPP_VERSION=12.3.0.116 NVJPEG_VERSION=12.3.2.38 CUDNN_VERSION=9.1.0.70+cuda12.4 TRT_VERSION=10.1.0.27+cuda12.4.1.003 TRTOSS_VERSION=24.06 NSIGHT_SYSTEMS_VERSION=2024.2.3.38 NSIGHT_COMPUTE_VERSION=2024.2.0.16 DALI_VERSION=1.38.0 DALI_BUILD=15028468 POLYGRAPHY_VERSION=0.49.11 TRANSFORMER_ENGINE_VERSION=1.7 MODEL_OPT_VERSION=0.11.2 /bin/sh -c echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf && echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf # buildkit
13.39KB 2024-06-04 01:08:59 ADD docs.tgz / # buildkit
0.00B 2024-06-04 01:08:59 ENV DALI_VERSION=1.38.0 DALI_BUILD=15028468 POLYGRAPHY_VERSION=0.49.11 TRANSFORMER_ENGINE_VERSION=1.7 MODEL_OPT_VERSION=0.11.2
0.00B 2024-06-04 01:08:59 ARG MODEL_OPT_VERSION
0.00B 2024-06-04 01:08:59 ARG TRANSFORMER_ENGINE_VERSION
0.00B 2024-06-04 01:08:59 ARG POLYGRAPHY_VERSION
0.00B 2024-06-04 01:08:59 ARG DALI_BUILD
0.00B 2024-06-04 01:08:59 ARG DALI_VERSION
0.00B 2024-06-04 01:08:59 LABEL com.nvidia.nccl.version=2.21.5 com.nvidia.cublas.version=12.5.2.13 com.nvidia.cufft.version=11.2.3.18 com.nvidia.curand.version=10.3.6.39 com.nvidia.cusparse.version=12.4.1.24 com.nvidia.cusolver.version=11.6.2.40 com.nvidia.cutensor.version=2.0.1.2 com.nvidia.npp.version=12.3.0.116 com.nvidia.nvjpeg.version=12.3.2.38 com.nvidia.cudnn.version=9.1.0.70+cuda12.4 com.nvidia.tensorrt.version=10.1.0.27+cuda12.4.1.003 com.nvidia.tensorrtoss.version=24.06 com.nvidia.nsightsystems.version=2024.2.3.38 com.nvidia.nsightcompute.version=2024.2.0.16
4.91GB 2024-06-04 01:08:59 RUN |17 CUDA_VERSION=12.5.0.023 CUDA_DRIVER_VERSION=555.42.02 JETPACK_HOST_MOUNTS= NCCL_VERSION=2.21.5 CUBLAS_VERSION=12.5.2.13 CUFFT_VERSION=11.2.3.18 CURAND_VERSION=10.3.6.39 CUSPARSE_VERSION=12.4.1.24 CUSOLVER_VERSION=11.6.2.40 CUTENSOR_VERSION=2.0.1.2 NPP_VERSION=12.3.0.116 NVJPEG_VERSION=12.3.2.38 CUDNN_VERSION=9.1.0.70+cuda12.4 TRT_VERSION=10.1.0.27+cuda12.4.1.003 TRTOSS_VERSION=24.06 NSIGHT_SYSTEMS_VERSION=2024.2.3.38 NSIGHT_COMPUTE_VERSION=2024.2.0.16 /bin/sh -c /nvidia/build-scripts/installLIBS.sh && /nvidia/build-scripts/installCUDNN.sh && /nvidia/build-scripts/installTRT.sh && /nvidia/build-scripts/installNSYS.sh && /nvidia/build-scripts/installNCU.sh && /nvidia/build-scripts/installCUTENSOR.sh && if [ -z "${JETPACK_HOST_MOUNTS}" ]; then /nvidia/build-scripts/installNCCL.sh; fi; # buildkit
0.00B 2024-05-17 11:27:23 ENV NCCL_VERSION=2.21.5 CUBLAS_VERSION=12.5.2.13 CUFFT_VERSION=11.2.3.18 CURAND_VERSION=10.3.6.39 CUSPARSE_VERSION=12.4.1.24 CUSOLVER_VERSION=11.6.2.40 CUTENSOR_VERSION=2.0.1.2 NPP_VERSION=12.3.0.116 NVJPEG_VERSION=12.3.2.38 CUDNN_VERSION=9.1.0.70+cuda12.4 TRT_VERSION=10.1.0.27+cuda12.4.1.003 TRTOSS_VERSION=24.06 NSIGHT_SYSTEMS_VERSION=2024.2.3.38 NSIGHT_COMPUTE_VERSION=2024.2.0.16
0.00B 2024-05-17 11:27:23 ARG NSIGHT_COMPUTE_VERSION
0.00B 2024-05-17 11:27:23 ARG NSIGHT_SYSTEMS_VERSION
0.00B 2024-05-17 11:27:23 ARG TRTOSS_VERSION
0.00B 2024-05-17 11:27:23 ARG TRT_VERSION
0.00B 2024-05-17 11:27:23 ARG CUDNN_VERSION
0.00B 2024-05-17 11:27:23 ARG NVJPEG_VERSION
0.00B 2024-05-17 11:27:23 ARG NPP_VERSION
0.00B 2024-05-17 11:27:23 ARG CUTENSOR_VERSION
0.00B 2024-05-17 11:27:23 ARG CUSOLVER_VERSION
0.00B 2024-05-17 11:27:23 ARG CUSPARSE_VERSION
0.00B 2024-05-17 11:27:23 ARG CURAND_VERSION
0.00B 2024-05-17 11:27:23 ARG CUFFT_VERSION
0.00B 2024-05-17 11:27:23 ARG CUBLAS_VERSION
0.00B 2024-05-17 11:27:23 ARG NCCL_VERSION
0.00B 2024-05-17 11:27:23 LABEL com.nvidia.volumes.needed=nvidia_driver com.nvidia.cuda.version=9.0
0.00B 2024-05-17 11:27:23 ENV _CUDA_COMPAT_PATH=/usr/local/cuda/compat ENV=/etc/shinit_v2 BASH_ENV=/etc/bash.bashrc SHELL=/bin/bash NVIDIA_REQUIRE_CUDA=cuda>=9.0
58.91KB 2024-05-17 11:27:23 RUN |3 CUDA_VERSION=12.5.0.023 CUDA_DRIVER_VERSION=555.42.02 JETPACK_HOST_MOUNTS= /bin/sh -c cp -vprd /nvidia/. / && patch -p0 < /etc/startup_scripts.patch && rm -f /etc/startup_scripts.patch # buildkit
463.24MB 2024-05-17 11:27:23 RUN |3 CUDA_VERSION=12.5.0.023 CUDA_DRIVER_VERSION=555.42.02 JETPACK_HOST_MOUNTS= /bin/sh -c /nvidia/build-scripts/installCUDA.sh # buildkit
0.00B 2024-05-17 11:28:43 RUN |3 CUDA_VERSION=12.5.0.023 CUDA_DRIVER_VERSION=555.42.02 JETPACK_HOST_MOUNTS= /bin/sh -c if [ -n "${JETPACK_HOST_MOUNTS}" ]; then echo "/usr/lib/aarch64-linux-gnu/tegra" > /etc/ld.so.conf.d/nvidia-tegra.conf && echo "/usr/lib/aarch64-linux-gnu/tegra-egl" >> /etc/ld.so.conf.d/nvidia-tegra.conf; fi # buildkit
0.00B 2024-05-17 11:28:43 ENV CUDA_VERSION=12.5.0.023 CUDA_DRIVER_VERSION=555.42.02 CUDA_CACHE_DISABLE=1 NVIDIA_REQUIRE_JETPACK_HOST_MOUNTS=
0.00B 2024-05-17 11:28:43 ARG JETPACK_HOST_MOUNTS
0.00B 2024-05-17 11:28:43 ARG CUDA_DRIVER_VERSION
0.00B 2024-05-17 11:28:43 ARG CUDA_VERSION
321.35MB 2024-05-17 11:27:01 RUN /bin/sh -c export DEBIAN_FRONTEND=noninteractive && apt-get update && apt-get install -y --no-install-recommends apt-utils build-essential ca-certificates curl libncurses5 libncursesw5 patch wget rsync unzip jq gnupg libtcmalloc-minimal4 # buildkit
0.00B 2024-04-27 21:18:37 /bin/sh -c #(nop) CMD ["/bin/bash"]
77.86MB 2024-04-27 21:18:37 /bin/sh -c #(nop) ADD file:a5d32dc2ab15ff0d7dbd72af26e361eb1f3e87a0d29ec3a1ceab24ad7b3e6ba9 in /
0.00B 2024-04-27 21:18:35 /bin/sh -c #(nop) LABEL org.opencontainers.image.version=22.04
0.00B 2024-04-27 21:18:35 /bin/sh -c #(nop) LABEL org.opencontainers.image.ref.name=ubuntu
0.00B 2024-04-27 21:18:35 /bin/sh -c #(nop) ARG LAUNCHPAD_BUILD_ARCH
0.00B 2024-04-27 21:18:35 /bin/sh -c #(nop) ARG RELEASE

镜像信息 

{
    "Id": "sha256:7598d7ac994bf5ac43a5550c2ea4d35a7bb0aad419288d82ff2401a83752620d",
    "RepoTags": [
        "nvcr.io/nvidia/pytorch:24.06-py3",
        "swr.cn-north-4.myhuaweicloud.com/ddn-k8s/docker.io/nvcr.io/nvidia/pytorch:24.06-py3"
    ],
    "RepoDigests": [
        "nvcr.io/nvidia/pytorch@sha256:d2e356ae415541b0088ea8a6883a9c620a411974a451b7e368ce88dff2a74f6c",
        "swr.cn-north-4.myhuaweicloud.com/ddn-k8s/docker.io/nvcr.io/nvidia/pytorch@sha256:57732c285e690d5e24f564a16625e0c25b78e64757ce6538e0a611d751527df5"
    ],
    "Parent": "",
    "Comment": "buildkit.dockerfile.v0",
    "Created": "2024-06-07T01:17:53.042480121Z",
    "Container": "",
    "ContainerConfig": null,
    "DockerVersion": "",
    "Author": "",
    "Config": {
        "Hostname": "",
        "Domainname": "",
        "User": "",
        "AttachStdin": false,
        "AttachStdout": false,
        "AttachStderr": false,
        "ExposedPorts": {
            "6006/tcp": {},
            "8888/tcp": {}
        },
        "Tty": false,
        "OpenStdin": false,
        "StdinOnce": false,
        "Env": [
            "PATH=/usr/local/lib/python3.10/dist-packages/torch_tensorrt/bin:/usr/local/mpi/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin:/opt/tensorrt/bin",
            "CUDA_VERSION=12.5.0.023",
            "CUDA_DRIVER_VERSION=555.42.02",
            "CUDA_CACHE_DISABLE=1",
            "NVIDIA_REQUIRE_JETPACK_HOST_MOUNTS=",
            "_CUDA_COMPAT_PATH=/usr/local/cuda/compat",
            "ENV=/etc/shinit_v2",
            "BASH_ENV=/etc/bash.bashrc",
            "SHELL=/bin/bash",
            "NVIDIA_REQUIRE_CUDA=cuda\u003e=9.0",
            "NCCL_VERSION=2.21.5",
            "CUBLAS_VERSION=12.5.2.13",
            "CUFFT_VERSION=11.2.3.18",
            "CURAND_VERSION=10.3.6.39",
            "CUSPARSE_VERSION=12.4.1.24",
            "CUSOLVER_VERSION=11.6.2.40",
            "CUTENSOR_VERSION=2.0.1.2",
            "NPP_VERSION=12.3.0.116",
            "NVJPEG_VERSION=12.3.2.38",
            "CUDNN_VERSION=9.1.0.70+cuda12.4",
            "TRT_VERSION=10.1.0.27+cuda12.4.1.003",
            "TRTOSS_VERSION=24.06",
            "NSIGHT_SYSTEMS_VERSION=2024.2.3.38",
            "NSIGHT_COMPUTE_VERSION=2024.2.0.16",
            "DALI_VERSION=1.38.0",
            "DALI_BUILD=15028468",
            "POLYGRAPHY_VERSION=0.49.11",
            "TRANSFORMER_ENGINE_VERSION=1.7",
            "MODEL_OPT_VERSION=0.11.2",
            "LD_LIBRARY_PATH=/usr/local/lib/python3.10/dist-packages/torch/lib:/usr/local/lib/python3.10/dist-packages/torch_tensorrt/lib:/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64",
            "NVIDIA_VISIBLE_DEVICES=all",
            "NVIDIA_DRIVER_CAPABILITIES=compute,utility,video",
            "NVIDIA_PRODUCT_NAME=PyTorch",
            "GDRCOPY_VERSION=2.3.1-1",
            "HPCX_VERSION=2.19",
            "MOFED_VERSION=5.4-rdmacore39.0",
            "OPENUCX_VERSION=1.17.0",
            "OPENMPI_VERSION=4.1.7",
            "RDMACORE_VERSION=39.0",
            "OPAL_PREFIX=/opt/hpcx/ompi",
            "OMPI_MCA_coll_hcoll_enable=0",
            "LIBRARY_PATH=/usr/local/cuda/lib64/stubs:",
            "PYTORCH_BUILD_VERSION=2.4.0a0+f70bd71",
            "PYTORCH_VERSION=2.4.0a0+f70bd71",
            "PYTORCH_BUILD_NUMBER=0",
            "NVIDIA_PYTORCH_VERSION=24.06",
            "NVFUSER_BUILD_VERSION=f73ff1b",
            "NVFUSER_VERSION=f73ff1b",
            "PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python",
            "NVPL_LAPACK_MATH_MODE=PEDANTIC",
            "PYTHONIOENCODING=utf-8",
            "LC_ALL=C.UTF-8",
            "PIP_DEFAULT_TIMEOUT=100",
            "NVM_DIR=/usr/local/nvm",
            "NODE_OPTIONS=--openssl-legacy-provider",
            "JUPYTER_PORT=8888",
            "TENSORBOARD_PORT=6006",
            "UCC_CL_BASIC_TLS=^sharp",
            "TORCH_CUDA_ARCH_LIST=5.2 6.0 6.1 7.0 7.2 7.5 8.0 8.6 8.7 9.0+PTX",
            "PYTORCH_HOME=/opt/pytorch/pytorch",
            "CUDA_HOME=/usr/local/cuda",
            "TORCH_ALLOW_TF32_CUBLAS_OVERRIDE=1",
            "USE_EXPERIMENTAL_CUDNN_V8_API=1",
            "COCOAPI_VERSION=2.0+nv0.8.0",
            "TORCH_CUDNN_V8_API_ENABLED=1",
            "CUDA_MODULE_LOADING=LAZY",
            "NVIDIA_BUILD_ID=96418707"
        ],
        "Cmd": null,
        "Image": "",
        "Volumes": null,
        "WorkingDir": "/workspace",
        "Entrypoint": [
            "/opt/nvidia/nvidia_entrypoint.sh"
        ],
        "OnBuild": null,
        "Labels": {
            "com.nvidia.build.id": "96418707",
            "com.nvidia.build.ref": "3017efeca4038be2c8c0f965cf868f48051bff16",
            "com.nvidia.cublas.version": "12.5.2.13",
            "com.nvidia.cuda.version": "9.0",
            "com.nvidia.cudnn.version": "9.1.0.70+cuda12.4",
            "com.nvidia.cufft.version": "11.2.3.18",
            "com.nvidia.curand.version": "10.3.6.39",
            "com.nvidia.cusolver.version": "11.6.2.40",
            "com.nvidia.cusparse.version": "12.4.1.24",
            "com.nvidia.cutensor.version": "2.0.1.2",
            "com.nvidia.nccl.version": "2.21.5",
            "com.nvidia.npp.version": "12.3.0.116",
            "com.nvidia.nsightcompute.version": "2024.2.0.16",
            "com.nvidia.nsightsystems.version": "2024.2.3.38",
            "com.nvidia.nvjpeg.version": "12.3.2.38",
            "com.nvidia.pytorch.version": "2.4.0a0+f70bd71",
            "com.nvidia.tensorrt.version": "10.1.0.27+cuda12.4.1.003",
            "com.nvidia.tensorrtoss.version": "24.06",
            "com.nvidia.volumes.needed": "nvidia_driver",
            "org.opencontainers.image.ref.name": "ubuntu",
            "org.opencontainers.image.version": "22.04"
        }
    },
    "Architecture": "amd64",
    "Os": "linux",
    "Size": 19150140892,
    "GraphDriver": {
        "Data": {
            "LowerDir": "/var/lib/docker/overlay2/30eb939b687afa0bd012ea1206da9459edc4306ef71ecded7a752a986b2e25e8/diff:/var/lib/docker/overlay2/ba554f39712ec087f7837a1c18586febd107aba299bde265c85198f258fbd584/diff:/var/lib/docker/overlay2/0d164d14b25b75759a96daf0d0322188a54e692bf58c68e229d489ce7d78bceb/diff:/var/lib/docker/overlay2/91da6c0e7591ad25a5f33d0007ac753e1687f39c4aaf82dd4e847092c64eb5be/diff:/var/lib/docker/overlay2/9213914227cf7626741c171e821c372b4555846c79297d12776ce290215e775e/diff:/var/lib/docker/overlay2/8f7d19b2ad8f87f3496c857b2ec5eb79fd91b787301f15e2355669f48a1bf340/diff:/var/lib/docker/overlay2/39e1cbed216f8d3306d50ea7d74a2e276fd12b197d82e6d51f34659abb8c15d9/diff:/var/lib/docker/overlay2/2882d174de617f9cde9877b509d9f5e8f3df61e4db71562ce71509fa2070e78c/diff:/var/lib/docker/overlay2/7591d6abab3c65653df622e25f4a5ce781aa826d78427e79805dde94d0858e37/diff:/var/lib/docker/overlay2/0331adb3e2a06541d3be04e958b92c0b5ccdd72de94ead45de8a4221eca77a3d/diff:/var/lib/docker/overlay2/32fbda253d709372bf0894a5aa04f3f898084e68af9ab0a267c518bd9ba1b82d/diff:/var/lib/docker/overlay2/1d3f7db2b50bf038e94e6dc2acc1dfbd6d063cb3ff7a7cb44f18765aaffe554c/diff:/var/lib/docker/overlay2/88caacb82aacb48336838c13db45e6cf3b5dae320372218b8aa46e91d63c2e6c/diff:/var/lib/docker/overlay2/fa17c4a5b84b336a883a141aaa7ecf6b325baabca5aa0f59311dfc07140d9bb2/diff:/var/lib/docker/overlay2/5966d55b335c8f4e6ec030bb5e1fe41034e404b69bbc7b85f9a85f9e7af3606a/diff:/var/lib/docker/overlay2/247df15c5e4d29ecbccef6634e49cd8e725d0eaa45525d0766473d998677d586/diff:/var/lib/docker/overlay2/150c93cfef9336df14d56f40f065aea4a9981fb0c61644d0381d5b0fc4b36e37/diff:/var/lib/docker/overlay2/9ed769d1d5a41bd897119bd98c89d69579cdb4fd6853201e0af6f187085f1dfc/diff:/var/lib/docker/overlay2/24e207c85de3e9bd3fb79c197118198abd4d52756e7d9de42a52ca7c5768ed1a/diff:/var/lib/docker/overlay2/ea5c64d0669307f7843d241ef5044ada40e3fff2b220eafa41e22ba3be143d8d/diff:/var/lib/docker/overlay2/3910a5a6a02d2488e816f4c221a2c4ffbd079d4181ab9f80f42fc14d623f9568/diff:/var/lib/docker/overlay2/03587c695cd264ac131655c37dc990c41b76a0d0c7a5b648f163601ad9d31cf4/diff:/var/lib/docker/overlay2/4157263e188fa7896eb3218cd8b9c593314b30d9bd0e6999fc7130f8f7215523/diff:/var/lib/docker/overlay2/c316585f258ee903036a0fc65019342f04a1d3eeb6a0b9f6eaa8e6d03cd8a77d/diff:/var/lib/docker/overlay2/1975cbddb9f547a5172bb626f15d36747fe4dc4520be90130202b3e6827d55b3/diff:/var/lib/docker/overlay2/abb77c1ef4b8a075c6c098c56a188c4e60c714ae41a2d6cc7bcb3ed578591625/diff:/var/lib/docker/overlay2/6f45aa5d169c33e32201c6dfc6a9f45554051b18be99a55212c09f22c0abd984/diff:/var/lib/docker/overlay2/64b5fee90e1700562c9e7998f5137d74bf45589c0313c415b7dc504c285f2a9f/diff:/var/lib/docker/overlay2/6157e88b1a352845df64a7f2fa19438a9044f1f7ab04a3ae4a5f5f2f31a3465a/diff:/var/lib/docker/overlay2/25c5b5f6253e90c3f44c5b961b3f145e25ed62fa1fd641aa43ab3de00452c800/diff:/var/lib/docker/overlay2/d6e041c7b0e8bce5cbbe1edd7855b4535b0090fc88115a75b69277049659a32b/diff:/var/lib/docker/overlay2/7731abb3b959d1c462c345f65c4de4712c7b1cd8ea280dd23f6380a6bb32b7f3/diff:/var/lib/docker/overlay2/cabbc53d112756d38a311d9daafe3292b0d265d17b1f52306ed096e1c31165f3/diff:/var/lib/docker/overlay2/275730d6b11a865cecc31d23e78913228784457f92b1bbdf69509d5f587b6252/diff:/var/lib/docker/overlay2/7da7a3525984e57ef51c8c85ae0272a42290c4736b6aa56e5592b2e51da23bd4/diff:/var/lib/docker/overlay2/79a7ad3f2f7c6330ba3844aed8ab84255dd7d49ade5a4e731ad00bd46ac39fbb/diff:/var/lib/docker/overlay2/95f106ab729810888a2732b0f72a47a4bacdd8ce699c0dfc6116bf44d00ed7d9/diff:/var/lib/docker/overlay2/5264c64d19ac09c09564790d8ceced44ddf3b4a918d1f9af96f19164caec7f82/diff:/var/lib/docker/overlay2/434f5e704957a3fd10caf1d6e013ad672bd587c044f79785230f5e9b660b4cd6/diff:/var/lib/docker/overlay2/a88ae650aa337abbe610c2fdb2671f6863da55956fd42c53f3832cdaf79a3133/diff:/var/lib/docker/overlay2/555202f702369741d2ff532973b5e42e7e936d9a36dbdff4723e6ea57e82d624/diff:/var/lib/docker/overlay2/807d88e56dbf4e44fddfc933e55477c7ed50a688394a51e72c52b28dff1307bc/diff:/var/lib/docker/overlay2/4db493137635783aea84288f71f3c4cea6227513c40ad854f570fa17a5227fc9/diff:/var/lib/docker/overlay2/d9e8fbe9855824da4706d61b40773e1968b5fc30990f5e60b46287298adff236/diff:/var/lib/docker/overlay2/6d755388029b23e8042b0685c361f91567809bf9d0530b4c9a9cd1f214cdac35/diff:/var/lib/docker/overlay2/452c5be669f5dc02ec0fe1c396a36452d6b8aa0f7685aa2193cdde73c527cc0d/diff:/var/lib/docker/overlay2/02beaacdceac5270dd51c1b8eb9349bdf13715d0fb85ca6fadb6224342992350/diff:/var/lib/docker/overlay2/c50913517f433bcc0c66c2d281cd3e794e5a099d1148a32b164acf62aea27d25/diff:/var/lib/docker/overlay2/da6e206c322d7ff954ba075f8cd4d41b5935761106a94f812f65f190d6221288/diff",
            "MergedDir": "/var/lib/docker/overlay2/798dd80efa81f124886ce0ab73b06f072be59da7c492adeea4ac6996fdc1f19c/merged",
            "UpperDir": "/var/lib/docker/overlay2/798dd80efa81f124886ce0ab73b06f072be59da7c492adeea4ac6996fdc1f19c/diff",
            "WorkDir": "/var/lib/docker/overlay2/798dd80efa81f124886ce0ab73b06f072be59da7c492adeea4ac6996fdc1f19c/work"
        },
        "Name": "overlay2"
    },
    "RootFS": {
        "Type": "layers",
        "Layers": [
            "sha256:629ca62fb7c791374ce57626d6b8b62c76378be091a0daf1a60d32700b49add7",
            "sha256:a0f3c9de4531d6d96a6193315d926ca652e08bd6f369043fd503c9cab74ccc21",
            "sha256:5f70bf18a086007016e948b04aed3b82103a36bea41755b6cddfaf10ace3c6ef",
            "sha256:09d18c0049e80998367672ee527294acb4f4d0a98fb123d60f525607832cb0c7",
            "sha256:bcf76e404a611e0c5a86706140835b21fc15e83ae07514942a7ea366ba7fac75",
            "sha256:7008e5cf3af07cd5d846575cccd1e000b4a3f4a5558c782901d12c91eb802c65",
            "sha256:3e6372480b0ece2d17729c78791a19ebf6597a905eaa64da241c56a4cf479486",
            "sha256:93fa5b2ce12a53a5d74943bd5b0c5c514ec95f88986876691454682995b596ea",
            "sha256:3e88d5b8e4c95b27b577a21cc7f5363b2003221fb129ca0300e37b94e4d6ec36",
            "sha256:03b84bdb3c56578e3ca86d080b29c610f141e0c7bd928fe612e303ddfbece34d",
            "sha256:67bd7641e3fcc8511e5bb6095877634a62bab743f1b23ee7c47ac034e7bb3a0a",
            "sha256:e7d2572f31d9e4c1f2ec0169d66d601887fc25494c2c2e7f0bd5e85274c32afb",
            "sha256:5252e80a3c1a11e872e90ae2c444c34b86f1f7facb36e12f7ef620fde6c03269",
            "sha256:648e7ef3cd75904a62bf57ccfbf839c0f5934c35b4319cd3f3a1875cd16ca305",
            "sha256:a88d15def64ab6f059d56443607ae98d07477e6b891a8d68bbd15659e40a0f5e",
            "sha256:4e69442b7c5239cb838542aaab7be18f3945d8c7f995fee626d3c6bd4a22ff14",
            "sha256:adff7c2691066a30b14a5d1642ffe9c71cf637e81aae1df48f1f7de381e2758f",
            "sha256:83658509c9646e8609ae55d5af7243c522a053044bb585b69d5bd2af1d032bc6",
            "sha256:5f70bf18a086007016e948b04aed3b82103a36bea41755b6cddfaf10ace3c6ef",
            "sha256:b1e3f9dc34a0fe38bdab0342a1b514d9dc8d89f79a2a823e1a63e4039756174a",
            "sha256:baf7bb74f9843ea8161336d2f55a5f3e9ada85e79068efa62c51d6e9f244dd46",
            "sha256:9c8e4abb154cf1f619eb2da98fc32c96e8ae13c72094108b756f4ad0c77201c0",
            "sha256:82361bb6f8fb40b561af6cae3d2705156301107d8716f50603206e5073e820b0",
            "sha256:7d1e9fce2aa06da43b3250df3f8652c1daa6f9890394223117e0bb5ea84a4f75",
            "sha256:604fd16a9c3f6aad1c963e6fe35bf6bda69d39309a49e9eeec69c1cf235c76ec",
            "sha256:531a69c6d0c9bd8633e90cc0f5d7eb8e42fb416355951adfd35c50a6064ec552",
            "sha256:ba5cd1abb1f1142c40810fe48b0c81ff69280fd5e14823fc88c6391ab4e39631",
            "sha256:86b3b78b4c10ede81d05c649fd061e5bd7e4aad17065ce812ded9131fc3ad9ac",
            "sha256:f07cf957eb833ccd177a8006d29301cb88cdcc88e5335c5c57ed66070113fb8c",
            "sha256:33fd3f130c0d68e12a07b45fb1d6db1189d7d2e277a16faada5f40790448b05c",
            "sha256:c5cdf4d54f5974454dcdcb58b019a56bbb3d33e629b89d531553303359fdca90",
            "sha256:b39423a6c4a4711a1cfb4b974c52944e7ea5f3d4c616f51832c1afdafab9b678",
            "sha256:999a3a60a948b07d0c984d27c3f6323161efae8d0ad2dba143a8a3615439405c",
            "sha256:5a407c998b5d5fffdc7450127bbc102d8a05270a9f54b65b5afd23491a18601c",
            "sha256:1d0f45bff1ae73c92ffe097c12598e0c80f0a15045e070b490883944a0828091",
            "sha256:1c487b07721c5ffce6de45cc1ae289d446229488fac471c6565ba144c0b2edfd",
            "sha256:5f70bf18a086007016e948b04aed3b82103a36bea41755b6cddfaf10ace3c6ef",
            "sha256:5ae77878e133c6589afcca88d38dfd97d2192e0e58129d1b236bd4e57032454d",
            "sha256:a6191815787829ba2d55f40f2c71d0f11c6c2c0afca5fc7f112abcaccc68d9c9",
            "sha256:0a2267caf6c9ae9fffba6c73dfaa218bd492b67c8cdd9e15b71f2ddf8c4d7f43",
            "sha256:22cfdd6ebbec8e8ae0a8f0b968a70873108700ced2462419e3eef7372478dc54",
            "sha256:2cb092d15b407237f18ed65fc623ac71cc55ce9573550752f84d3e16260427a9",
            "sha256:b3187f1ffe782d89cc7c3a78ae565f4d8823280dfe177a027fc11c8b84442259",
            "sha256:a80d9e0549e8aff2f339bff72552ecbd58c0c752ed4b4575bc75a60178895def",
            "sha256:b81909895bc48008b35420c3294b3afbdecd7b882b03845a2899bfb93e9d77f0",
            "sha256:beacc1cc62ae5df39d1f8fe0cea3090d905330a60b216297fc1a5761c56ec43a",
            "sha256:26f8ff834cf2e9ef9c2a9034156dd10f4b058cb86ef4886881d2a108f73a9682",
            "sha256:2ad36b0ec6c760aeb0bbef9e486c1c5cc47d4eff997db80453005dcd8ee3b7ea",
            "sha256:21e95f7357637cc6336438bdb01f51bc2eb2c2be074d2540db59fe24888737e7",
            "sha256:f6f8d946223990205e6e5a2ea2a386933697662bcdc3733bea0007dc87300591"
        ]
    },
    "Metadata": {
        "LastTagTime": "2024-10-22T00:08:50.340799976+08:00"
    }
}

更多版本