- 692 名前:オーバーテクナナシー [2021/05/06(木) 10:00:06.92 ID:s3m0gTnp.net]
- >>690
> 1ヶ月前に3080で動かした おめ! >CPU版と比べて同じコードだとエラーが出た https://qiita.com/Lirimy/items/2706018dcd8fc1fc9e61 #dtype = torch.FloatTensor # CPU dtype = torch.cuda.FloatTensor # GPU tensor = torch.Tensor(...).type(dtype) こんな感じで切り替えか でもこれと違うやり方を提示しているサイトが2つあるなあ どっちを選ぶか
|
|