diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-12 11:29:02 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-12 11:29:02 +0800 |
| commit | 966d4431c037b0c4641aa2a5fc22f05be064b331 (patch) | |
| tree | 0239ba89d31857a7f086acf627fc1bbf167855a9 /utils/misc.py | |
| parent | 7825f978f198e56958703f0d08f7ccbd8cef49ca (diff) | |
| parent | 36cf502afe9b93efe31c244030270b0a62e644b8 (diff) | |
Merge branch 'master' into python3.8
# Conflicts:
# models/model.py
Diffstat (limited to 'utils/misc.py')
| -rw-r--r-- | utils/misc.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/utils/misc.py b/utils/misc.py new file mode 100644 index 0000000..b850830 --- /dev/null +++ b/utils/misc.py @@ -0,0 +1,10 @@ +import os + +from utils.configuration import SystemConfiguration + + +def set_visible_cuda(config: SystemConfiguration): + """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 |