今までGeForce GTX 1060でPytorchを使って勉強していましたが、メモリ不足に悩まされる事から
メモリ容量が大きくてディープラーニング的にコスパの良いグラボを探してコレにしました。
が、RTX 3060に交換した途端Pytorchが動かなくなっててんやわんやしたので。

はじめに

タイトルの通りですが、RTX3060でのPytorh環境を作るのにてんやわんやしたので備忘録として。

要点

上記の環境に記載のバージョン、または動作する組み合わせにてインストールする
CUDAは11.x系をインストールしてください。

こちらの記事で知りました。
PyTorchでRTX 3090を使う(Qiita)

私の場合はこの手順で動作するようになりました。
どなたかのお役に立てるよう、またグラボの高騰が終息しますよう・・・。