Chooses gpu:0 or the first GPU in a GPU cluster.
CUDA_VISIBLE_DEVICES=0 python deeplab/model_test.pyMore info about this environment variable: https://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html#env-vars
Last active
June 15, 2018 06:34
-
-
Save rlan/7d4053d54a90012e4d73be55d62ae52f to your computer and use it in GitHub Desktop.
Choose GPU via command line
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment