![](/img/trans.png)
[英]torch.manual_seed(seed) get RuntimeError: CUDA error: device-side assert triggered
[英]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 中獲取 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.
請檢查device_count()
的 output。 您可以使用超出范圍的索引。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.