PyTorch | NVIDIA NGC

NVIDIA

PyTorch

Container

NVIDIA

PyTorch

PyTorch is a GPU accelerated tensor computational framework. Functionality can be extended with common Python libraries such as NumPy and SciPy. Automatic differentiation is done with a tape-based system at the functional and neural network layer levels.

NVIDIA AI Enterprise Supported

Layer	Label		Created
43f2bbea75f5544417a94c58dec0daaba8ec6dade6f6ccc3a439a96daffb6849	CONFIG	`Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp`	01/19/2025 10:09 AM UTC
8fde8b8807147cb1b768f9008c6dc2f06f9d2b6f5dea19ad580c980c5f030f4e	ARG	`NVIDIA_BUILD_REF=3857cfd7a27acea1d4e19243329fcd1dae3b222e`	01/19/2025 10:09 AM UTC
5d0774eb078a3c45e24a549fdbdc88cf00756941668a7b0292b72e63c196e2d0	LABEL	`com.nvidia.build.id=134983853`	01/19/2025 10:09 AM UTC
3d9da0bc0184c01697845498762889d900f2c26c83e0a457b05626fd64d53da5	ENV	`NVIDIA_BUILD_ID=134983853`	01/19/2025 10:09 AM UTC
055e1afb9c0811f0b0d335cd028f585f45c436b01db24c5ea5f27f93c8d4a0a6	ARG	`NVIDIA_BUILD_ID=134983853`	01/19/2025 10:09 AM UTC
bf38dc5da80a563686773fd336ecb9139b0027248b12b5f7e36404a2516dd45f	COPY	`entrypoint.d/ /opt/nvidia/entrypoint.d/`	01/19/2025 10:09 AM UTC
51cabe2118dbd828d6625f8831972278363d8daa720e7ea5b5a0197b0e750560	RUN	`NVIDIA_PYTORCH_VERSION=25.01 PYTORCH_BUILD_VERSION=2.6.0a0+ecf3bae NVFUSER_BUILD_VERSION=6627725 TARGETARCH=amd64 PYVER=3.12 PYVER_MAJMIN=312 L4T=0 DALI_EXTRA_INDEX_URL=http://sqrl/nvdl/datasets/dali/pip-dali 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`	01/19/2025 10:09 AM UTC
d85bd606acd28100cc156efa1e0995293e1fc1dbcdc277f73010d84b4d78ff10	ENV	`TORCH_NCCL_USE_COMM_NONBLOCKING=0`	01/19/2025 10:09 AM UTC
7d8895b76c086ea530384300d95fd20fc023514b8f585979cf0891bb53b0654b	ENV	`CUDA_MODULE_LOADING=LAZY`	01/19/2025 10:09 AM UTC
13a00dce5c62fcc1617239fa1bbc3f64f94a3aca429328da265f149d80dde203	ENV	`TORCH_CUDNN_V8_API_ENABLED=1`	01/19/2025 10:09 AM UTC