[ISSUE]: Hardware requirements / nvfuser for aoc_fruit_detector?

[KatherineJames]

Description

When using the launch file, I get the error below. I can recreate the error by simply running python and importing pytorch, and trying to access the gpu. See below. What are the hardware requirements for using aoc_fruit_detector?

import torch generates the following:
==== importing nvfuser failed ====
try run patch-nvfuser if https://github.com/NVIDIA/Fuser is installed via pip package

print(torch.cuda.is_available())

/usr/local/lib/python3.10/dist-packages/torch/cuda/init.py:107: UserWarning: CUDA initialization: Unexpected error from cudaGetDeviceCount(). Did you run some cuda functions before calling NumCudaDevices() that might have already set an error? Error 804: forward compatibility was attempted on non supported HW (Triggered internally at ../c10/cuda/CUDAFunctions.cpp:109.)

[KatherineJames]

To resolve inside container:

• pip install nvfuser-cu118-torch24
• export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib/python3.10/dist-packages/torch/lib:$LD_LIBRARY_PATH

This should potentially be added to the devcontainer setup.

Ensure host has cuda driver >= 520