removing alpine distribution

deepanker13 · Jan 5, 2024 · 5d22ccd · 5d22ccd
1 parent b11a534
commit 5d22ccd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sdk/python/Dockerfile.conformance b/sdk/python/Dockerfile.conformance
@@ -13,7 +13,7 @@
 # limitations under the License.
 
 # Dockerfile for building the source code of conformance tests
-FROM python:3.10-alpine
+FROM python:3.11
 
 WORKDIR /kubeflow/training-operator/sdk/python
 
@@ -25,6 +25,6 @@ COPY sdk/python/conformance/run.sh .
 RUN chmod +x run.sh
 
 RUN pip install pytest
-RUN python -m pip install -e .
+RUN python -m pip install -e . --extra-index-url https://download.pytorch.org/whl/cu117
 
 ENTRYPOINT [ "./run.sh" ]