← Back to all projectsGPU, HPC & Infrastructure
ML on GPU & HPC — Train VGG16/ResNet18 Under Slurm with CUDA & PyTorch
Training real AI models (VGG16, ResNet18) under Slurm using CUDA, PyTorch, and MPI — full HPC job lifecycle from submission to distributed checkpointing and evaluation.
PyTorchSlurmCUDAVGG16ResNet18HPC
View source on GitHub