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
1226014d4f472ea7f1a0a6a55a8b0dfa7d777408545e0b6bf410ace536270ac7	CONFIG	`Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp`	12/15/2022 3:12 AM UTC
b17ce6839ebf07e8000be5b40270cfe978c0187fc3de186628c1fc48c72bc96f	ARG	`NVIDIA_BUILD_REF`	12/15/2022 3:12 AM UTC
f6e535ddac4a578102baaa0f654c775c4088c905126b34ac1e77e2c64fde41ea	LABEL	`com.nvidia.build.id=49968248`	12/15/2022 3:12 AM UTC
13b1f077d669600e5ca518eb3810c5db93770d3de045c826d3c7edf06bc5348c	ENV	`NVIDIA_BUILD_ID=49968248`	12/15/2022 3:12 AM UTC
9e00d3558a81b3e5415b37e0b5a60c06b5f91a9ced97f1eb4ec4fd89edef0d6b	ARG	`NVIDIA_BUILD_ID`	12/15/2022 3:12 AM UTC
b1fa60ad910fe2f532587858f54f8f1ac3c812e9446a300041175e0a7df9fb06	COPY	`entrypoint.d/ /opt/nvidia/entrypoint.d/`	12/15/2022 3:12 AM UTC
35cb4aac91811bbce9316f0ce26da021b7a91897c161ee374498959a6db17dd5	RUN	`PYVER=3.8 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`	12/15/2022 3:12 AM UTC
027a88ad2695ddece94d363c5c60a144e7b92935ac9940385845532f7213a9fc	ENV	`CUDA_MODULE_LOADING=LAZY`	12/15/2022 3:12 AM UTC
66eaa186b8d6ed436289217230a93e6701ad5581a5753ebaa2ca04162e5e4cb8	ENV	`TORCH_CUDNN_V8_API_ENABLED=1`	12/15/2022 3:12 AM UTC
82a67d0d7b309ccddb042752bf19efd9272a14377e4de0b7bc3d95ee9c087520	RUN	`PYVER=3.8 pip install --upgrade git+https://github.com/NVIDIA/TransformerEngine.git@release_v${TRANSFORMER_ENGINE_VERSION}`	12/15/2022 3:12 AM UTC