PyTorchでCUDA error: no kernel image is available for execution on the device

CUDAを11.1に更新した後にPyTorchを実行した際に表題のエラー。

CUDA10だと通常の pip install torch torchvision torchaudio で良いようだが、11.1はバージョン違いによるエラーが起きた。

公式サイトにあるコマンドメーカーから得られるコマンドを使うことでエラーが消え、PyTorchを実行することが出来た。

pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

参考

Start Locally

Start Locally

pytorch.org