普通にPIPでインストールしたら以下のエラーがでた。
Could not load library libcudnn_ops_infer.so.8. Error: libcublas.so.11: cannot open shared object file: No such file or directory
公式のページを見て再インストールする。
pytorch.org
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118