jamesjg / FoodSAM

FoodSAM: Any Food Segmentation
Apache License 2.0
134 stars 13 forks source link

FoodSAM Colab GPU Version problems mmcv-full #4

Open JonaspythonG opened 9 months ago

JonaspythonG commented 9 months ago

Hello, I am trying to use FoodSAM in Google Colab but I am encountering a problem. As soon as I attempt to install mmcv-full and allocate GPU resources to the notebook, I receive an error named "building wheels". I did some research and found out that if I install using !pip install torch==1.12.0+cu116 torchvision==0.13.0+cu116 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu116, I can at least install mmcv versions 2.0.0 and 2.1.0. However, unfortunately, these versions do not have the Dictation function... Does anyone know a solution?

error: subprocess-exited-with-error

× python setup.py bdist_wheel did not run successfully. │ exit code: 1 ╰─> See above for output.

note: This error originates from a subprocess, and is likely not a problem with pip. Building wheel for mmcv-full (setup.py) ... error ERROR: Failed building wheel for mmcv-full Running setup.py clean for mmcv-full Failed to build mmcv-full ERROR: Could not build wheels for mmcv-full, which is required to install pyproject.toml-based project

ImportError: cannot import name 'DictAction' from 'mmcv.utils'

Queeny23333 commented 5 months ago

The same Q..

bastianMO commented 3 months ago

I solved this by downgrading torch version to 1.8.1+cu111 torchvision == 0.9.1+cu111, and use CUDA 11.1

eichi7 commented 3 months ago

@bastianMO can you share with us what you have done please