ImportError: libcudart.so.10.1: cannot open shared object file: No such file or directory

[bring-nirachornkul]

ImportError: libcudart.so.10.1: cannot open shared object file: No such file or directory

Here is my installation detail :

Torch version: 1.9.1+cu111
CUDA version: 11.1
PyTorch3D version: 0.3.0
PyTorch Lightning version: 1.5.7
Hydra Core version: 1.1.1
WandB version: 0.13.1

Here is my nvcc-version :

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Tue_May__3_18:49:52_PDT_2022
Cuda compilation tools, release 11.7, V11.7.64
Build cuda_11.7.r11.7/compiler.31294372_0

My spec :

CPU : AMD Ryzen 9 5900X 12-Core Processor
GPU : RTX 3090
RAM : 64 GBs
Linux OS : Ubuntu 23.04

Here is the error when I run train.py :

(v2a) blinkdrive@blinkdrive-System-Product-Name:~/Documents/Projects/Summer2024/02vid2avatar/code$ python train.py
Traceback (most recent call last):
  File "train.py", line 1, in <module>
    from v2a_model import V2AModel
  File "/home/blinkdrive/Documents/Projects/Summer2024/02vid2avatar/code/v2a_model.py", line 3, in <module>
    from lib.model.v2a import V2A
  File "/home/blinkdrive/Documents/Projects/Summer2024/02vid2avatar/code/lib/model/v2a.py", line 4, in <module>
    from .deformer import SMPLDeformer
  File "/home/blinkdrive/Documents/Projects/Summer2024/02vid2avatar/code/lib/model/deformer.py", line 4, in <module>
    from pytorch3d import ops
  File "/home/blinkdrive/miniconda3/envs/v2a/lib/python3.7/site-packages/pytorch3d/ops/__init__.py", line 5, in <module>
    from .graph_conv import GraphConv
  File "/home/blinkdrive/miniconda3/envs/v2a/lib/python3.7/site-packages/pytorch3d/ops/graph_conv.py", line 8, in <module>
    from pytorch3d import _C
ImportError: libcudart.so.10.1: cannot open shared object file: No such file or directory

Here is the error when I run test.py :

(v2a) blinkdrive@blinkdrive-System-Product-Name:~/Documents/Projects/Summer2024/02vid2avatar/code$ python test.py
Traceback (most recent call last):
  File "test.py", line 1, in <module>
    from v2a_model import V2AModel
  File "/home/blinkdrive/Documents/Projects/Summer2024/02vid2avatar/code/v2a_model.py", line 3, in <module>
    from lib.model.v2a import V2A
  File "/home/blinkdrive/Documents/Projects/Summer2024/02vid2avatar/code/lib/model/v2a.py", line 4, in <module>
    from .deformer import SMPLDeformer
  File "/home/blinkdrive/Documents/Projects/Summer2024/02vid2avatar/code/lib/model/deformer.py", line 4, in <module>
    from pytorch3d import ops
  File "/home/blinkdrive/miniconda3/envs/v2a/lib/python3.7/site-packages/pytorch3d/ops/__init__.py", line 5, in <module>
    from .graph_conv import GraphConv
  File "/home/blinkdrive/miniconda3/envs/v2a/lib/python3.7/site-packages/pytorch3d/ops/graph_conv.py", line 8, in <module>
    from pytorch3d import _C
ImportError: libcudart.so.10.1: cannot open shared object file: No such file or directory