it looks --index-url will override the original one

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Jan 29, 2024 · 75dc2a2 · 75dc2a2
1 parent 3e13cba
commit 75dc2a2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/source/install/docker/Dockerfile b/source/install/docker/Dockerfile
@@ -6,7 +6,7 @@ RUN python -m venv /opt/deepmd-kit
 ENV PATH="/opt/deepmd-kit/bin:$PATH"
 # Install package
 COPY dist /dist
-RUN if [ "${CUDA_VERSION}" = 11 ]; then export PIP_INDEX_URL=https://download.pytorch.org/whl/cu118; fi \
+RUN if [ "${CUDA_VERSION}" = 11 ]; then pip install torch --index-url https://download.pytorch.org/whl/cu118; fi \
     && pip install "$(ls /dist/deepmd_kit${VARIANT}-*manylinux*_x86_64.whl)[gpu,cu${CUDA_VERSION},lmp,ipi,torch]" \
     && dp -h \
     && lmp -h \