はじめに
pytorchをgithubからインストールすることに成功したものの、実行する際、以下のエラーが出てしまいました。この解決策を備忘録として残します。
(原因はよくわかっていません)
$ python test.py
ImportError: libcudart.so.10.1: cannot open shared object file: No such file or directory
問題
私の環境(変更前)は以下です。
- cuda 11.1
- Python : 3.8
- torch : 1.9.1 (cu111)
- pytorch3d : 0.3.0
※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を再インストールしたところ、無事動作しました。
環境(変更後)
- cuda 11.1
- Python : 3.8
- torch : 1.9.0 (cu111)
- pytorch3d : 0.7.8
おわりに
今回は、PyTorch3dを使用したコードを実行する際に生じるエラーの解決策を書きました。
torchのバージョンを1.9.1から1.9.0へダウングレードすることで、無事動作しましたが、原因はよくわかっていません。
torchとpytorch3dの対応表や、他の解決法があれば、ご教授いただければ幸いです。