torch geometric and cluster in dockerfile

Dockerfile

@@ -41,6 +41,7 @@ RUN pyenv install ${PYTHON_VERSION} && \
     pip install --no-cache-dir -U pip setuptools wheel
 
 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
+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
 COPY --chown=1000 requirements.txt /tmp
 RUN pip install --no-cache-dir -r /tmp/requirements.txt
 

requirements.txt

@@ -1,6 +1,4 @@
 pytorch-lightning>=2.1
-torch-geometric>=2
-torch-cluster>=1.6.3 -f https://data.pyg.org/whl/torch-2.1.0+cu121.html
 scikit-learn-intelex>=2024.1.0
 numpy>=1.26.4
 scikit-image>=0.22.0