Fix detectron2 build on HF: disable build isolation, add PyTorch to requirements, add apt packages, limit MAX_JOBS
Browse files- mvp.py +2 -1
- packages.txt +7 -0
- requirements.txt +3 -0
mvp.py
CHANGED
|
@@ -46,7 +46,8 @@ try:
|
|
| 46 |
except Exception:
|
| 47 |
pass
|
| 48 |
|
| 49 |
-
os.
|
|
|
|
| 50 |
REPO_ROOT = os.path.dirname(os.path.abspath(__file__))
|
| 51 |
sys.path.append(os.path.join(REPO_ROOT, "vggt"))
|
| 52 |
MK_PATH = os.path.join(REPO_ROOT, "MaskClustering")
|
|
|
|
| 46 |
except Exception:
|
| 47 |
pass
|
| 48 |
|
| 49 |
+
os.environ.setdefault("MAX_JOBS", "1")
|
| 50 |
+
os.system("python -m pip install --no-build-isolation -e ./MaskClustering/third_party/detectron2/")
|
| 51 |
REPO_ROOT = os.path.dirname(os.path.abspath(__file__))
|
| 52 |
sys.path.append(os.path.join(REPO_ROOT, "vggt"))
|
| 53 |
MK_PATH = os.path.join(REPO_ROOT, "MaskClustering")
|
packages.txt
CHANGED
|
@@ -1,4 +1,11 @@
|
|
| 1 |
build-essential
|
| 2 |
python3-dev
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
ninja-build
|
| 4 |
|
|
|
|
| 1 |
build-essential
|
| 2 |
python3-dev
|
| 3 |
+
cmake
|
| 4 |
+
ninja-build
|
| 5 |
+
ffmpeg
|
| 6 |
+
libgl1
|
| 7 |
+
libglib2.0-0
|
| 8 |
+
build-essential
|
| 9 |
+
python3-dev
|
| 10 |
ninja-build
|
| 11 |
|
requirements.txt
CHANGED
|
@@ -1,3 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
| 1 |
torch==2.3.1
|
| 2 |
torchvision==0.18.1
|
| 3 |
numpy==1.26.1
|
|
|
|
| 1 |
+
torch==2.1.2
|
| 2 |
+
torchvision==0.16.2
|
| 3 |
+
torchaudio==2.1.2
|
| 4 |
torch==2.3.1
|
| 5 |
torchvision==0.18.1
|
| 6 |
numpy==1.26.1
|