PyTorchのインストール

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をインストールできたので上出来である。

コメントする

このサイトはスパムを低減するために Akismet を使っています。コメントデータの処理方法の詳細はこちらをご覧ください