Resource
V-Net is a convolutional neural network for 3D image segmentation.
Use the NGC CLI to download:
Copied!
The following section lists the requirements in order to start training the V-Net Medical model.
Requirements
This repository contains a Dockerfile which extends the TensorFlow NGC container and encapsulates some additional dependencies. Aside from these dependencies, ensure you have the following components:
- TensorFlow 20.06-tf1-py3 NGC container
- GPU-based architecture:
For more information about how to get started with NGC containers, see the following sections from the NVIDIA GPU Cloud Documentation and the Deep Learning DGX Documentation: