Dockerfile needs to uninstall cython before cupy

ROCm · Jan 18, 2024 · 760ae80 · 760ae80
1 parent a613192
commit 760ae80
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tools/ci_build/github/pai/rocm-ci-pipeline-env.Dockerfile b/tools/ci_build/github/pai/rocm-ci-pipeline-env.Dockerfile
@@ -90,6 +90,7 @@ RUN wget https://download.open-mpi.org/release/open-mpi/v4.1/openmpi-4.1.5.tar.b
     rm openmpi-4.1.5.tar.bz2
 
 # Install CuPy, No stable version is available
+RUN pip uninstall -y cython
 RUN git clone https://github.com/ROCmSoftwarePlatform/cupy && cd cupy && \
     git checkout 432a8683351d681e00903640489cb2f4055d2e09 && \
     export CUPY_INSTALL_USE_HIP=1 && \