PyTorch-Jetson-Nano
PyTorch-Jetson-Nano copied to clipboard
Jetson Nano, Jetpack 4.6
Hi Team Thank you for your efforts to build and make the torch files available!
I want to know two things
- Are the torch files built using CUDA? I had installed the default torch provided by nvidia it doesnt use CUDA.
- I am getting an error as below when i install any of the 4 torch files (1.7 to 1.10).
"import torch; print(torch.__version__)"
Traceback (most recent call last):
File "<string>", line 1, in <module>
File "venv/lib/python3.6/site-packages/torch/__init__.py", line 196, in <module>
_load_global_deps()
File "venv/lib/python3.6/site-packages/torch/__init__.py", line 149, in _load_global_deps
ctypes.CDLL(lib_path, mode=ctypes.RTLD_GLOBAL)
File "/usr/lib/python3.6/ctypes/__init__.py", line 348, in __init__
self._handle = _dlopen(self._name, mode)
OSError: libcurand.so.10: cannot open shared object file: No such file or directory
Do you know what could be causing the issue? Thank your for your help in advance!
First question: Yes, we compile the wheel with CUDA and cuDNN acceleration.
The second issue can be a nightmare. We've downloaded our PyTorch 1.10 wheel and installed it on a clean machine to see if we can reproduce the error. However, it doesn't give any problem.
On the other hand, when googling your error, we get some hits. Please see for instance this topic.