liuhengyue/set_cuda_device.py

Last active August 23, 2018 21:08

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/liuhengyue/b27c736e575775691ddfcaa24d59db39.js"></script>
Save liuhengyue/b27c736e575775691ddfcaa24d59db39 to your computer and use it in GitHub Desktop.

Download ZIP

Select GPU

Raw

set_cuda_device.py

	import os
	os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
	os.environ["CUDA_VISIBLE_DEVICES"] = ""

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