We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Hi, could I add some code at the beginning of train.py to choose which GPU to use? Like set the os.environ.
The text was updated successfully, but these errors were encountered:
As you said, the GPU to use can be set by means of CUDA_VISIBLE_DEVICES environment variable.
e.g. (Inside script) os.environ['CUDA_VISIBLE_DEVICES'] = "0"
or
(outside script) export CUDA_VISIBLE_DEVICES="0"
Later, I will update the code to set GPU by tf.ConfigProto as another option.
Sorry, something went wrong.
Is cpu ok?
No branches or pull requests
Hi, could I add some code at the beginning of train.py to choose which GPU to use?
Like set the os.environ.
The text was updated successfully, but these errors were encountered: