/cuda-tests

Primary LanguagePython

cuda-tests

check-nvidia-cuda

Checks if libraries for cuda, cudart, and cudnn are installed. Checks if nvcc is installed, and runs it to show the version number. Checks if nvidia-smi is installed, and runs it to show the Nvidia driver and cuda version numbers.

check-gpu-with-torch

Creates, trains, and tests a neural network on the MNIST dataset.