Resource
NVIDIA Deep Learning Examples
GPUNet for PyTorchGPUNet is a new family of Convolutional Neural Networks crafted by NVIDIA AI.
21.12.1
Copied!
The following section lists the requirements you need to meet to start training the GPUNet.
Requirements
This repository contains a Dockerfile that extends the PyTorch NGC container and encapsulates all dependencies. You also need the following components to get started:
- NVIDIA Docker
- ImageNet 1K dataset
- Supported GPUs:
For more information about how to get started with NGC containers, refer tothe following sections from the NVIDIA GPU Cloud Documentation and the Deep Learning DGX Documentation: