marciopuga/test.sh

Last active July 15, 2019 00:36

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/marciopuga/633ac2b39214bd6faff2ce0b6a0693a9.js"></script>
Save marciopuga/633ac2b39214bd6faff2ce0b6a0693a9 to your computer and use it in GitHub Desktop.

Download ZIP

Test if Tensorflow inside a docker container can see nvidia-driver installed in the host

Raw

test.sh

docker run --runtime=nvidia -it --rm tensorflow/tensorflow:latest-gpu-py3  nvidia-smi

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment