/test_pytorch_gpu

Primary LanguagePythonMIT LicenseMIT

Utility Script

Description

This script is to test pytorch + GPU installation on any new machine. It will check if GPU is available and run a 10 epoch training on CIFAR10 dataset.

Install

run command pip install test-pytorch-gpu

Use

run command run_cifar