Cupyが使えないので、代わりにPyTorchでGPU計算を試みる。
PyTorchのインストールで苦戦したので、以下にメモしておく。
1.前提
自分のPC環境は以下の通り。
・GPUはRX2060
・OSはWindows11
・CUDA ToolkitとCUDAのバージョンは10.1
2.問題
PyTorchのサイトに行くと、下図のようにCUDAバージョン10.1がない。
3.対策
INSTALLING PREVIOUS VERSIONS OF PYTORCHでCUDAバージョン10.1を探し、
そのコマンドをコピーして使う。下図はその実行例。
4.参考資料
下記のWEBページにお世話になりました。ありがとうございます。
・PyTorchとCUDA Toolkitをインストールする
<個人的な感想>
CUDAバージョンを最新版に更新しようとも思ったが、更に泥沼にハマりそうなので
今動いているバージョンで何とかPyTorchをインストールできたので上出来である。