perler commited on
Commit
f198681
1 Parent(s): 3990bc7

try pyg install command from docu

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -1
Dockerfile CHANGED
@@ -41,10 +41,12 @@ RUN pyenv install ${PYTHON_VERSION} && \
41
  pip install --no-cache-dir -U pip setuptools wheel
42
 
43
  RUN pip install --no-cache-dir -U torch==2.1.1+cu121 torchvision>=0.16+cu121 --extra-index-url https://download.pytorch.org/whl/cu121
44
- RUN pip install --no-cache-dir -U torch-geometric>=2+cu121 torch-cluster>=1.6.3+cu121 --extra-index-url https://data.pyg.org/whl/torch-2.1.0+cu121.html
45
  COPY --chown=1000 requirements.txt /tmp
46
  RUN pip install --no-cache-dir -r /tmp/requirements.txt
47
 
 
 
48
  COPY --chown=1000 . ${HOME}/app
49
  RUN git clone https://github.com/cg-tuwien/ppsurf.git
50
  RUN cd ppsurf
 
41
  pip install --no-cache-dir -U pip setuptools wheel
42
 
43
  RUN pip install --no-cache-dir -U torch==2.1.1+cu121 torchvision>=0.16+cu121 --extra-index-url https://download.pytorch.org/whl/cu121
44
+ RUN pip install --no-cache-dir -U pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.1.0+cu121.html
45
  COPY --chown=1000 requirements.txt /tmp
46
  RUN pip install --no-cache-dir -r /tmp/requirements.txt
47
 
48
+ RUN pip list
49
+
50
  COPY --chown=1000 . ${HOME}/app
51
  RUN git clone https://github.com/cg-tuwien/ppsurf.git
52
  RUN cd ppsurf