We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b0c83db commit 9990179Copy full SHA for 9990179
monai/utils/misc.py
@@ -235,8 +235,7 @@ def set_determinism(
235
if seed is None:
236
# cast to 32 bit seed for CUDA
237
seed_ = torch.default_generator.seed() % (np.iinfo(np.int32).max + 1)
238
- if not torch.cuda._is_in_bad_fork():
239
- torch.cuda.manual_seed_all(seed_)
+ torch.manual_seed(seed_)
240
else:
241
seed = int(seed) % MAX_SEED
242
torch.manual_seed(seed)
0 commit comments