我在 google colab 上运行我的代码。 我收到 RuntimeError: CUDA 错误：设备端断言已触发。错误显示为 torch.manual_seed(seed)

Question

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 中获取 RuntimeError Traceback (most recent call last) 的错误。 manual_seed_all(种子) 8 tokenizer = 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.

Answer 1

请检查device_count()的 output。 您可以使用超出范围的索引。