You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Gerber, Mike 59b78549f7 | 4 years ago | |
---|---|---|
2019-12-tensorflow2-keras-cnn+lstm | 5 years ago | |
2020-03-tensorflow-vs-tensorflow-gpu | 5 years ago | |
assets | 4 years ago | |
README.md | 4 years ago | |
run | 4 years ago | |
run-docker | 4 years ago | |
test-nvidia | 4 years ago |
README.md
Test Nvidia environment in relation to TensorFlow
./run
tests the native system. One of tf1 or tf2 is expected to have no GPU available due to CUDA library incompatibility./run-docker
tests Docker support. Both TensorFlow versions should work as we're using a base image compatible to the respective version.
Example output
% ./run-docker
== tf1
GPU 0: GeForce RTX 2080 (UUID: GPU-612ce75c-1340-772b-039c-2a83a3ea5c95)
TensorFlow 1.15.3
GPU available: True
== tf2
GPU 0: GeForce RTX 2080 (UUID: GPU-612ce75c-1340-772b-039c-2a83a3ea5c95)
TensorFlow 2.3.0
GPU available: True