Torch 在 Jetson Nano 上运行 YOLOv5 时出错

Question

我正在尝试使用 Jetson Nano（Ubuntu 20.04 和 Python 3.8）GPU 运行 YOLOv5。我已经使用这些命令设置了 YOLOv5

git clone https://github.com/ultralytics/yolov5
cd yolov5
pip install -r requirements.txt

然后我尝试运行 detect.py 文件（使用基本权重和测试数据），我得到了这个错误：

Traceback (most recent call last):
  File "detect.py", line 37, in <module>
    import torch
  File "/home/jetson/.local/lib/python3.8/site-packages/torch/__init__.py", line 198, in <module>
    _load_global_deps()
  File "/home/jetson/.local/lib/python3.8/site-packages/torch/__init__.py", line 151, 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: libcublas.so.11: cannot open shared object file: No such file or directory