我在 google colab 上运行我的代码。正在收到 RuntimeError: CUDA error: device-side assert denied.Error 显示为 torch.manual_seed(seed)

原文标题 ：I am running my code on google colab. Am getting RuntimeError: CUDA error: device-side assert triggered.Error is showing for torch.manual_seed(seed)

import random
total_steps = 1
seed = 42
random.seed(seed)
np.random.seed(seed)
torch.manual_seed(seed)

以下是 ()4 random.seed(seed)5 np.random.seed(seed)—-> 6 torch.manual_seed(seed)7 torch.cuda.manual_seed_all 中的错误 am gettingRuntimeError Traceback (most recent call last) (种子)8 标记器 = BertTokenizer.from_pretrained(‘bert-base-uncased’)

3 frames
/usr/local/lib/python3.7/dist-packages/torch/cuda/random.py in cb()
    109         for i in range(device_count()):
    110             default_generator = torch.cuda.default_generators[i]
--> 111             default_generator.manual_seed(seed)
    112 
    113     _lazy_call(cb, seed_all=True)

RuntimeError: CUDA error: device-side assert triggered
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.