/test_cuda

A bash script for testing if cuda was properly installed on a GPU server by installing pytorch and check access via torch.cuda.is_available().

Primary LanguageShell

test_cuda

A bash script for testing if cuda was properly installed on a GPU server by installing pytorch and check access via torch.cuda.is_available().

Just run:

bash gpu_test.sh