いい加減、何度もハマっているので、忘れないように記録を残しておくようにします。
PyTorchをインストールするとき
- Anaconda NavigatorのGUIからPyTorchをインストールしないこと!
- 必ず https://pytorch.org/ を見ること!
そして、「INSTALL PYTORCH」のところで、自分の環境にあった選択をし、「Run this Command:」の文字列をコピペして実行すること。
今までの失敗
- torchvisionがない
- 直接condaでtorchvisionをインストールする
- PyTorchとバージョンが合わない
- PyTorchがCUDAでコンパイルされていない
- 直接condaでcudatoolkitをインストールする
- バージョンが合わない
- 直接condaでcudatoolkitをインストールする
- 直接condaでtorchvisionをインストールする
おまけ
- torchsummaryはpipでインストールする