From 4495e10e5159ee54814e73ae7bfb27db2754ae69 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Tue, 12 Jan 2021 20:23:40 +0800 Subject: Add Google TPU support --- utils/misc.py | 9 --------- 1 file changed, 9 deletions(-) delete mode 100644 utils/misc.py (limited to 'utils/misc.py') diff --git a/utils/misc.py b/utils/misc.py deleted file mode 100644 index c047618..0000000 --- a/utils/misc.py +++ /dev/null @@ -1,9 +0,0 @@ -import os -from typing import Dict - - -def set_visible_cuda(config: Dict): - """Set environment variable CUDA device(s)""" - CUDA_VISIBLE_DEVICES = config.get('CUDA_VISIBLE_DEVICES', None) - if CUDA_VISIBLE_DEVICES: - os.environ['CUDA_VISIBLE_DEVICES'] = CUDA_VISIBLE_DEVICES -- cgit v1.2.3