import torch
print(torch.version)
→1.3.1
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
print(device)
→cuda:0
x = torch.randn(10)
x.to(device)
print(x.is_cuda)
→False
google colab上でこのコードを実行しましたが、xがGPUを認識してくれません。
どうすればxにGPUを認識させられるのでしょうか?
(→は実行結果です)
あなたの回答
tips
プレビュー