Skip to content

Instantly share code, notes, and snippets.

@ltupin
Created August 30, 2018 08:45
Show Gist options
  • Select an option

  • Save ltupin/36d53bb75b53e92d144f499db0f05bf4 to your computer and use it in GitHub Desktop.

Select an option

Save ltupin/36d53bb75b53e92d144f499db0f05bf4 to your computer and use it in GitHub Desktop.
Some commands to check nvidia support inside docker
python --version
nvcc --version
nvidia-smi
import sys
print(sys.version_info)
import tensorflow as tf; print(tf.__version__)
import tensorflow as tf; print(tf.GIT_VERSION, tf.VERSION)
import tensorflow as tf
if tf.test.gpu_device_name():
    print('Default GPU Device: {}'.format(tf.test.gpu_device_name()))
else:
    print("Please install GPU version of TF")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment