Skip to content

Commit

Permalink
fix dgl install (#32)
Browse files Browse the repository at this point in the history
Co-authored-by: Kaustubh Tangsali <[email protected]>
  • Loading branch information
ktangsali and ktangsali authored May 2, 2023
1 parent 61dcc2b commit 307e446
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -19,8 +19,8 @@ FROM nvcr.io/nvidia/pytorch:$PYT_VER-py3 as builder
RUN pip install --upgrade pip setuptools

# Install nightly build of dgl
RUN pip install --pre dgl -f https://data.dgl.ai/wheels/cu117/repo.html
RUN pip install --pre dglgo -f https://data.dgl.ai/wheels-test/repo.html
RUN pip install --no-deps --pre dgl -f https://data.dgl.ai/wheels/cu117/repo.html
RUN pip install --no-deps --pre dglgo -f https://data.dgl.ai/wheels-test/repo.html
ENV DGLBACKEND=pytorch

ENV _CUDA_COMPAT_TIMEOUT=90
Expand Down Expand Up @@ -80,4 +80,4 @@ FROM deploy as docs
# Install CI packages
RUN pip install tensorflow>=2.11.0 warp-lang>=0.6.0 protobuf==3.20.0
# Install packages for Sphinx build
RUN pip install recommonmark==0.7.1 sphinx==5.1.1 sphinx-rtd-theme==1.0.0 pydocstyle==6.1.1 nbsphinx==0.8.9 nbconvert==6.4.3 jinja2==3.0.3
RUN pip install recommonmark==0.7.1 sphinx==5.1.1 sphinx-rtd-theme==1.0.0 pydocstyle==6.1.1 nbsphinx==0.8.9 nbconvert==6.4.3 jinja2==3.0.3

0 comments on commit 307e446

Please sign in to comment.