Georg commited on
Commit
a586677
·
1 Parent(s): 3d1bba0

Optimized Docker build to fix OOM errors

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -16,7 +16,8 @@ ENV MAX_JOBS=1
16
  # Install nvdiffrast (CUDA rasterizer) - needs GPU, build here
17
  RUN git clone --depth 1 https://github.com/NVlabs/nvdiffrast.git /tmp/nvdiffrast \
18
  && cd /tmp/nvdiffrast \
19
- && python3 -m pip install . --no-use-pep517 --no-build-isolation --no-deps \
 
20
  && python3 -c "import nvdiffrast.torch" \
21
  && rm -rf /tmp/nvdiffrast
22
 
 
16
  # Install nvdiffrast (CUDA rasterizer) - needs GPU, build here
17
  RUN git clone --depth 1 https://github.com/NVlabs/nvdiffrast.git /tmp/nvdiffrast \
18
  && cd /tmp/nvdiffrast \
19
+ && python3 setup.py install \
20
+ && python3 - <<'PY'\nimport sysconfig\nfrom pathlib import Path\nsite = Path(sysconfig.get_paths()[\"purelib\"])\ndist = site / \"nvdiffrast-0.0.0.dist-info\"\ndist.mkdir(exist_ok=True)\n(dist / \"METADATA\").write_text(\"Metadata-Version: 2.1\\nName: nvdiffrast\\nVersion: 0.0.0\\n\")\n(dist / \"WHEEL\").write_text(\"Wheel-Version: 1.0\\nGenerator: manual\\nRoot-Is-Purelib: false\\nTag: py3-none-any\\n\")\n(dist / \"top_level.txt\").write_text(\"nvdiffrast\\n\")\n(dist / \"RECORD\").write_text(\"\")\nPY \
21
  && python3 -c "import nvdiffrast.torch" \
22
  && rm -rf /tmp/nvdiffrast
23