Q&A
以下の情報を追記できますでしょうか?
- OS (Windows or Linux?)
- CUDAバージョン
- torch.cuda.is_available()の結果
- device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu"); print(device)
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
失礼しました．
OS:ubuntu :ubuntu 18.04.4 LTS
Cuda: v9.0.176
torch.cuda.is_available(): true
device: cuda:0
以上となります．
jupyter labを使っており，一度終了して再び起動し，このカーネルのみを実行するとGPUでまわりました．どういう理由かわかりませんが．．．
あなたの回答
tips
プレビュー
