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.
run command pip install test-pytorch-gpu
run command run_cifar
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.
run command pip install test-pytorch-gpu
run command run_cifar
MIT License