Updated lookup of CUDA devices.

这个提交包含在:
Craig Warren
2019-03-08 16:58:49 +00:00
父节点 4a4226038c
当前提交 ff2a8b32b1

查看文件

@@ -388,6 +388,7 @@ def detect_check_gpus(deviceIDs):
# Get list of available GPU device IDs
deviceIDsavail = os.environ.get('CUDA_VISIBLE_DEVICES')
deviceIDsavail = [int(s) for s in deviceIDsavail.split(',')]
# Print information about all detected GPUs
gpus = []