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

torch geometric and cluster in dockerfile

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -0
  2. requirements.txt +0 -2
Dockerfile CHANGED
@@ -41,6 +41,7 @@ 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
  COPY --chown=1000 requirements.txt /tmp
45
  RUN pip install --no-cache-dir -r /tmp/requirements.txt
46
 
 
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
 
requirements.txt CHANGED
@@ -1,6 +1,4 @@
1
  pytorch-lightning>=2.1
2
- torch-geometric>=2
3
- torch-cluster>=1.6.3 -f https://data.pyg.org/whl/torch-2.1.0+cu121.html
4
  scikit-learn-intelex>=2024.1.0
5
  numpy>=1.26.4
6
  scikit-image>=0.22.0
 
1
  pytorch-lightning>=2.1
 
 
2
  scikit-learn-intelex>=2024.1.0
3
  numpy>=1.26.4
4
  scikit-image>=0.22.0