@@ -13,32 +13,33 @@ These examples, along with our NVIDIA deep learning software stack, are provided
13
13
14
14
15
15
## Computer Vision
16
- | Models | Framework | A100 | AMP | Multi-GPU | Multi-Node | TRT | ONNX | Triton | DLC | NB |
17
- | ------------- | ------------- | ------------- | ------------- | ------------- | ------------- | ------------- | ------------- | ------------- | ------------- | ------------- |
18
- | [ ResNet-50] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/resnet50v1.5 ) | PyTorch | Yes | Yes | Yes | - | Yes | - | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/triton/resnet50 ) | Yes | - |
19
- | [ ResNeXt-101] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/resnext101-32x4d ) | PyTorch | Yes | Yes | Yes | - | Yes | - | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/triton/resnext101-32x4d ) | Yes | - |
20
- | [ SE-ResNeXt-101] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/se-resnext101-32x4d ) | PyTorch | Yes | Yes | Yes | - | Yes | - | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/triton/se-resnext101-32x4d ) | Yes | - |
21
- | [ EfficientNet-B0] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/efficientnet ) | PyTorch | Yes | Yes | Yes | - | - | - | - | Yes | - |
22
- | [ EfficientNet-B4] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/efficientnet ) | PyTorch | Yes | Yes | Yes | - | - | - | - | Yes | - |
23
- | [ EfficientNet-WideSE-B0] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/efficientnet ) | PyTorch | Yes | Yes | Yes | - | - | - | - | Yes | - |
24
- | [ EfficientNet-WideSE-B4] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/efficientnet ) | PyTorch | Yes | Yes | Yes | - | - | - | - | Yes | - |
25
- | [ Mask R-CNN] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Segmentation/MaskRCNN ) | PyTorch | Yes | Yes | Yes | - | - | - | - | - | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/blob/master/PyTorch/Segmentation/MaskRCNN/pytorch/notebooks/pytorch_MaskRCNN_pyt_train_and_inference.ipynb ) |
26
- | [ nnUNet] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Segmentation/nnUNet ) | PyTorch | Yes | Yes | Yes | - | - | - | - | Yes | - |
27
- | [ SSD] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Detection/SSD ) | PyTorch | Yes | Yes | Yes | - | - | - | - | - | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/blob/master/PyTorch/Detection/SSD/examples/inference.ipynb ) |
28
- | [ GPUNet] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/GPUNet ) | PyTorch | Yes | Yes | Yes | - | Yes | Yes | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/GPUNet/triton/ ) | Yes | - |
29
- | [ ResNet-50] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Classification/ConvNets/resnet50v1.5 ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | Yes | - |
30
- | [ ResNeXt101] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Classification/ConvNets/resnext101-32x4d ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | Yes | - |
31
- | [ SE-ResNeXt-101] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Classification/ConvNets/se-resnext101-32x4d ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | Yes | - |
32
- | [ Mask R-CNN] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow2/Segmentation/MaskRCNN ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | Yes | - |
33
- | [ SSD] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Detection/SSD ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | Yes | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/blob/master/TensorFlow/Detection/SSD/models/research/object_detection/object_detection_tutorial.ipynb ) |
34
- | [ U-Net Ind] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Segmentation/UNet_Industrial ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | Yes | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Segmentation/UNet_Industrial/notebooks ) |
35
- | [ U-Net Med] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Segmentation/UNet_Medical ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | Yes | - |
36
- | [ U-Net 3D] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Segmentation/UNet_3D_Medical ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | Yes | - |
37
- | [ V-Net Med] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Segmentation/VNet ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | Yes | - |
38
- | [ U-Net Med] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow2/Segmentation/UNet_Medical ) | TensorFlow2 | Yes | Yes | Yes | - | - | - | - | Yes | - |
39
- | [ Mask R-CNN] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow2/Segmentation/MaskRCNN ) | TensorFlow2 | Yes | Yes | Yes | - | - | - | - | Yes | - |
40
- | [ EfficientNet] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow2/Classification/ConvNets/efficientnet ) | TensorFlow2 | Yes | Yes | Yes | Yes | - | - | - | Yes | - |
41
- | [ ResNet-50] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/MxNet/Classification/RN50v1.5 ) | MXNet | - | Yes | Yes | - | - | - | - | - | - |
16
+ | Models | Framework | A100 | AMP | Multi-GPU | Multi-Node | TRT | ONNX | Triton | DLC | NB |
17
+ | -------------------------------------------------------------------------------------------------------------------------------------| --------------| ------| ------------- | ------------- | ------------- | -----| ------------- | ------------- | ------------- | ------------- |
18
+ | [ EfficientNet-B0] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/efficientnet ) | PyTorch | Yes | Yes | Yes | - | - | - | - | Yes | - |
19
+ | [ EfficientNet-B4] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/efficientnet ) | PyTorch | Yes | Yes | Yes | - | - | - | - | Yes | - |
20
+ | [ EfficientNet-WideSE-B0] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/efficientnet ) | PyTorch | Yes | Yes | Yes | - | - | - | - | Yes | - |
21
+ | [ EfficientNet-WideSE-B4] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/efficientnet ) | PyTorch | Yes | Yes | Yes | - | - | - | - | Yes | - |
22
+ | [ EfficientNet] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow2/Classification/ConvNets/efficientnet ) | TensorFlow2 | Yes | Yes | Yes | Yes | - | - | - | Yes | - |
23
+ | [ GPUNet] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/GPUNet ) | PyTorch | Yes | Yes | Yes | - | Yes | Yes | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/GPUNet/triton/ ) | Yes | - |
24
+ | [ Mask R-CNN] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Segmentation/MaskRCNN ) | PyTorch | Yes | Yes | Yes | - | - | - | - | - | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/blob/master/PyTorch/Segmentation/MaskRCNN/pytorch/notebooks/pytorch_MaskRCNN_pyt_train_and_inference.ipynb ) |
25
+ | [ Mask R-CNN] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow2/Segmentation/MaskRCNN ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | Yes | - |
26
+ | [ Mask R-CNN] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow2/Segmentation/MaskRCNN ) | TensorFlow2 | Yes | Yes | Yes | - | - | - | - | Yes | - |
27
+ | [ nnUNet] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Segmentation/nnUNet ) | PyTorch | Yes | Yes | Yes | - | - | - | - | Yes | - |
28
+ | [ ResNet-50] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/MxNet/Classification/RN50v1.5 ) | MXNet | - | Yes | Yes | - | - | - | - | - | - |
29
+ | [ ResNet-50] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PaddlePaddle/Classification/RN50v1.5 ) | PaddlePaddle | Yes | Yes | Yes | - | Yes | - | - | - | - |
30
+ | [ ResNet-50] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/resnet50v1.5 ) | PyTorch | Yes | Yes | Yes | - | Yes | - | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/triton/resnet50 ) | Yes | - |
31
+ | [ ResNet-50] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Classification/ConvNets/resnet50v1.5 ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | Yes | - |
32
+ | [ ResNeXt-101] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/resnext101-32x4d ) | PyTorch | Yes | Yes | Yes | - | Yes | - | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/triton/resnext101-32x4d ) | Yes | - |
33
+ | [ ResNeXt-101] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Classification/ConvNets/resnext101-32x4d ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | Yes | - |
34
+ | [ SE-ResNeXt-101] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/se-resnext101-32x4d ) | PyTorch | Yes | Yes | Yes | - | Yes | - | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/triton/se-resnext101-32x4d ) | Yes | - |
35
+ | [ SE-ResNeXt-101] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Classification/ConvNets/se-resnext101-32x4d ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | Yes | - |
36
+ | [ SSD] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Detection/SSD ) | PyTorch | Yes | Yes | Yes | - | - | - | - | - | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/blob/master/PyTorch/Detection/SSD/examples/inference.ipynb ) |
37
+ | [ SSD] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Detection/SSD ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | Yes | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/blob/master/TensorFlow/Detection/SSD/models/research/object_detection/object_detection_tutorial.ipynb ) |
38
+ | [ U-Net Ind] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Segmentation/UNet_Industrial ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | Yes | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Segmentation/UNet_Industrial/notebooks ) |
39
+ | [ U-Net Med] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Segmentation/UNet_Medical ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | Yes | - |
40
+ | [ U-Net 3D] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Segmentation/UNet_3D_Medical ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | Yes | - |
41
+ | [ U-Net Med] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow2/Segmentation/UNet_Medical ) | TensorFlow2 | Yes | Yes | Yes | - | - | - | - | Yes | - |
42
+ | [ V-Net Med] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Segmentation/VNet ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | Yes | - |
42
43
43
44
## Natural Language Processing
44
45
| Models | Framework | A100 | AMP | Multi-GPU | Multi-Node | TRT | ONNX | Triton | DLC | NB |
0 commit comments