GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation. のエラー対処
GeForce RTX 3090がある新しいマシンでは以下のようなエラーが報告されている。
GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.
If you want to use the GeForce RTX 3090 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/
これはCUDA11.1でないから出るエラーのようだ。
なのでこちらからCUDA11.1のものを選び、
https://pytorch.org/get-started/previous-versions/
pip, condaでpytorchなどをインストールすれば上手くいく。
以下を参照
