drozdgk commited on
Commit
22b92ec
·
1 Parent(s): 2c86904

Fix detectron2 build on HF: disable build isolation, add PyTorch to requirements, add apt packages, limit MAX_JOBS

Browse files
Files changed (3) hide show
  1. mvp.py +2 -1
  2. packages.txt +7 -0
  3. requirements.txt +3 -0
mvp.py CHANGED
@@ -46,7 +46,8 @@ try:
46
  except Exception:
47
  pass
48
 
49
- os.system("python -m pip install -e ./MaskClustering/third_party/detectron2/")
 
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