diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-12 20:23:40 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-12 20:23:40 +0800 |
| commit | 4495e10e5159ee54814e73ae7bfb27db2754ae69 (patch) | |
| tree | ac2bd997784229d58b72188dfa32d1c1edd0aa57 /utils/misc.py | |
| parent | 0c027822a48440be59f9b4889831af7879ecf277 (diff) | |
Add Google TPU support
Diffstat (limited to 'utils/misc.py')
| -rw-r--r-- | utils/misc.py | 9 |
1 files changed, 0 insertions, 9 deletions
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 |