/deep-learning-singularity

Deep learning Singularity builds using TensorFlow GPU and CUDA drivers

Primary LanguageShell

Deep learning Singularity builds

Singularity builds using TensorFlow GPU and CUDA drivers both from scratch and from Docker containers (e.g. TensorFlow GPU Docker container).

Nvidia Docker build

  • check-tensorflow.py checks if TensorFlow is working and GPUs are available