Hello,
I have been working with YOLOv8 on my Jetson Orin NX for a couple of weeks, but executing everything on the CPU because cuda was not available eventhough I had it installed. Now, I am trying to solve this issue to be able to execute using the GPU and train my models faster, but I haven’t found the solution yet.
For context, I have JetPack 5.1.2, Python 3.8 and cuda 12.4. After some research, I found that the version of Pytorch that I needed to install in order to be compatible with JetPack was 2.1.0a. I uninstalled the previous version of pythorch and followed this tutorial below to install the new one.
After the installation, when I import torch, the output is:
Traceback (most recent call last):
File “/home/laura/.local/lib/python3.8/site-packages/torch/init.py”, line 168, in _load_global_deps
ctypes.CDLL(lib_path, mode=ctypes.RTLD_GLOBAL)
File “/usr/lib/python3.8/ctypes/init.py”, line 373, in init
self._handle = _dlopen(self._name, mode)
OSError: libcufft.so.10: cannot open shared object file: No such file or directory
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File “”, line 1, in
File “/home/laura/.local/lib/python3.8/site-packages/torch/init.py”, line 228, in
_load_global_deps()
File “/home/laura/.local/lib/python3.8/site-packages/torch/init.py”, line 189, in _load_global_deps
_preload_cuda_deps(lib_folder, lib_name)
File “/home/laura/.local/lib/python3.8/site-packages/torch/init.py”, line 154, in _preload_cuda_deps
raise ValueError(f"{lib_name} not found in the system path {sys.path}")
ValueError: libcublas.so.*[0-9] not found in the system path [‘’, ‘/usr/lib/python38.zip’, ‘/usr/lib/python3.8’, ‘/usr/lib/python3.8/lib-dynload’, ‘/home/laura/.local/lib/python3.8/site-packages’, ‘/usr/local/lib/python3.8/dist-packages’, ‘/usr/lib/python3/dist-packages’, ‘/usr/lib/python3.8/dist-packages’]
I made some research and I have found that this error could be related to libraries and the cuda version, but I’m not sure if installing another version would work, as I already tried with cuda 11.4, 12.4 and 12.6.
Just in case it’s useful, when I run the command $ls /usr/local/cuda-12.4/lib64 the output is:
Any idea about how to solve this would be welcomed.
Thank you in advance!