search

Rumarino-Team / hydrus-software-stack

Unified software stack for the Hydrus submarine that will include computer vision, embedded, and controls code.
1 stars 4 forks source link

Fix the docker compose file for amd64 without cuda. #23

Closed Cruiz102 closed 1 month ago

Cruiz102 commented 1 month ago

The current implementation downloads all the PyTorch CUDA dependencies inside the Docker container, even in environments where CUDA is not used. We need to update the dependencies so that PyTorch is installed with CPU-only support in these cases.

Cruiz102 commented 1 month ago

this thread is gold https://github.com/ultralytics/ultralytics/issues/2542

Cruiz102 commented 1 month ago

For future reference the solution to build the ultralytics for cpu only is to execute the command.

python3.9 -m pip install --extra-index-url https://download.pytorch.org/whl/cpu ultralytics