CUDA error: device-side assert triggeredについて

pytorchで画像解析をしています．
CPUでは動くのですが，GPU動作にすると件名のエラーがでます．このエラー表示のみで何が原因かわからないのですが，ご存知でしたら教えていただけると助かります．
よろしくお願いします．

python
1
2device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
3
4net.to(device)
5

エラー出力

python
1--->      net.to(device)
2RuntimeError: CUDA error: device-side assert triggered
3

yymmt

2020/07/11 16:50

以下の情報を追記できますでしょうか? - OS (Windows or Linux?) - CUDAバージョン - torch.cuda.is_available()の結果 - device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu"); print(device)

nigo1973

2020/07/12 05:53

失礼しました． OS:ubuntu :ubuntu 18.04.4 LTS Cuda: v9.0.176 torch.cuda.is_available(): true device: cuda:0 以上となります． jupyter labを使っており，一度終了して再び起動し，このカーネルのみを実行するとGPUでまわりました．どういう理由かわかりませんが．．．