GPUの使用方法について

Question

ノートパソコン付属GPUからPyTorchを動かす方法がわかりません。

私は今、GeForce MX150が内臓されたlatitudeというノートパソコンを使用しています。

このGPU上でPyTorchを動かしたいのですが、やり方がわかりません。

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(device)
#cpu

上記のようにcpu環境でしか動かず、GPUを認識してくれません。
どうすればよろしいのでしょうか？
どなたか詳しい方、いらっしゃいますでしょうか、、、？

Answer 1

私の場合ですがNvidiaドライバが合わず切り替わりませんでした

pytorchのtorch.cuda.is_available()がFalseを出す問題

では最新版となっていますが経験としては

NvidiaドライバとCUDAとcuDNNとTensorflow-gpuとPythonのバージョンの対応

のほうが良いかと思います。