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
0d509d34ba87bafc69f7a36029b6488a2a16169d1eea2ae6db071d5ed1331ed7	CONFIG	`Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp`	11/21/2023 12:31 AM UTC
e427f930c1073f346ef29284c4e28746d617fd1c0e869a9b804f99084bab84c8	ARG	`NVIDIA_BUILD_REF`	11/21/2023 12:31 AM UTC
f7d5df7b8e4b9d00140e44cc98ff804bc4d478ab6c7dd24adaa3dda23b1481ec	LABEL	`com.nvidia.build.id=74957752`	11/21/2023 12:31 AM UTC
382f7cdf8f762dedd80419fc717d9e2c298a17bc2652f8b404a3e886acb7dbc1	ENV	`NVIDIA_BUILD_ID=74957752`	11/21/2023 12:31 AM UTC
fc2806f75c273bfb3fbdcf2d10202fd552763023976cb17b581ad18d85906585	ARG	`NVIDIA_BUILD_ID`	11/21/2023 12:31 AM UTC
78370251ca59f072b2718f2093d03f23df1dacc703226aa07a6dd86cd051d6f8	COPY	`entrypoint.d/ /opt/nvidia/entrypoint.d/`	11/21/2023 12:31 AM UTC
eac9ff19528e544e2ea1458b10f29bbb8d0b2789406c413683527693f930554e	RUN	`PYVER=3.10 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`	11/21/2023 12:31 AM UTC
f58bc959e2e5d8fc67c280412c22e02876980a150ca5843c75e40db8789ada33	ENV	`CUDA_MODULE_LOADING=LAZY`	11/21/2023 12:31 AM UTC
8e68c9b03cea7236bbd11e27ac7a3e0e7b164ba3ea405ba856a28b4273b6ade8	ENV	`TORCH_CUDNN_V8_API_ENABLED=1`	11/21/2023 12:31 AM UTC
c8af5ae01c8575a58decd20d357834295ed15416883a65767afb5585f4089602	RUN	`PYVER=3.10 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`	11/21/2023 12:31 AM UTC