/tf-test

Quick test code to assist debugging compute servers

Primary LanguagePython

tf-test

Some test code you can check-out to test your GPU configuration

Dependencies

Code is intended for testing existing environments, so no requirements.txt, but all you need is:

  • numpy
  • tensorflow
  • tensorflow-gpu (if testing GPUs)

Test if GPU is detected

python check-gpu.py

Test if a simple dense-layered NN can run

python train-dense.py

Test if an mnist conv-layered NN can run

python train-conv.py

Code adapted from this medium article