From c250ac9f20068767e1e61268587d0430b6655efa Mon Sep 17 00:00:00 2001 From: Semen Zhydenko Date: Thu, 8 Nov 2018 11:13:41 +0100 Subject: [PATCH] divices -> devices --- mpi_util.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mpi_util.py b/mpi_util.py index 7da01dc..18ab942 100644 --- a/mpi_util.py +++ b/mpi_util.py @@ -43,9 +43,9 @@ def guess_available_gpus(n_gpus=None): if n_gpus is not None: return list(range(n_gpus)) if 'CUDA_VISIBLE_DEVICES' in os.environ: - cuda_visible_divices = os.environ['CUDA_VISIBLE_DEVICES'] - cuda_visible_divices = cuda_visible_divices.split(',') - return [int(n) for n in cuda_visible_divices] + cuda_visible_devices = os.environ['CUDA_VISIBLE_DEVICES'] + cuda_visible_devices = cuda_visible_devices.split(',') + return [int(n) for n in cuda_visible_devices] if 'RCALL_NUM_GPU' not in os.environ: n_gpus = int(os.environ['RCALL_NUM_GPU']) return list(range(n_gpus))