Repository navigation
CUDA_VISIBLE_DEVICES setting ignored after import monai #7407
Description
Describe the bug
CUDA_VISIBLE_DEVICES setting ignored after import monai
To Reproduce
import os
import torch
print('before import monai:', torch.cuda.device_count())
import monai
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
print('after import monai:')
torch.cuda.set_device(1) # should failed
Expected behavior
CUDA_VISIBLE_DEVICES setting should not ignored after import monai
Activity
fix Project-MONAI#7407
Fix CUDA_VISIBLE_DEVICES setting ignored (#7408)
Fix CUDA_VISIBLE_DEVICES setting ignored (Project-MONAI#7408)
Fix CUDA_VISIBLE_DEVICES setting ignored (Project-MONAI#7408)
Fix CUDA_VISIBLE_DEVICES setting ignored (Project-MONAI#7408)