はじめに

pytorchをgithubからインストールすることに成功したものの、実行する際、以下のエラーが出てしまいました。この解決策を備忘録として残します。
(原因はよくわかっていません)

$ python test.py
ImportError: libcudart.so.10.1: cannot open shared object file: No such file or directory

問題

私の環境(変更前)は以下です。

※pytorch3dは以下のコマンドでインストールしました

pip install git+https://github.com/facebookresearch/pytorch3d.git

cuda11.xを使用しているため、libcudart.so.10.1が見つからないようです。公式サイトより、cudaとtorchのバージョンに対応するものwheelファイルからインストールを試みましたが、同様のエラー。

pip install pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py38_cu111_pyt191/index.html

こちらでは、torchのバージョンによって未対応と言及されています。

torch1.9.xを使用する必要がどうしてもあったため、試しにtorch1.9.0(cu111)に変更し、pytorch3dを再インストールしたところ、無事動作しました。

環境(変更後)

今回は、PyTorch3dを使用したコードを実行する際に生じるエラーの解決策を書きました。
torchのバージョンを1.9.1から1.9.0へダウングレードすることで、無事動作しましたが、原因はよくわかっていません。

torchとpytorch3dの対応表や、他の解決法があれば、ご教授いただければ幸いです。