Skip to content

Commit 3628656

Browse files
authored
[VL] Cleanup the GPU docker script (#11482)
* Modify Dockerfile for CUDA toolkit and build process Updated Dockerfile to install CUDA toolkit 13-1 and streamline setup process. * Clean up Dockerfile by removing unnecessary files Remove Spark directory and ccache from Dockerfile.
1 parent a9d6abb commit 3628656

1 file changed

Lines changed: 13 additions & 10 deletions

File tree

dev/docker/cudf/Dockerfile

Lines changed: 13 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -16,23 +16,26 @@
1616
#
1717

1818
FROM ghcr.io/facebookincubator/velox-dev:adapters
19-
RUN yum install -y sudo patch maven perl && ln -sf /usr/local/bin/cmake /usr/bin
20-
21-
RUN git clone --depth=1 https://github.com/apache/incubator-gluten /opt/gluten
22-
23-
# Install spark to folder /opt
24-
RUN cd /opt/gluten/.github/workflows/util/ && ./install-spark-resources.sh 3.4
2519
ENV SPARK_HOME=/opt/spark-3.4.4-bin-hadoop3
2620
ENV PATH=$SPARK_HOME/bin:$PATH
2721
ENV CUDA_ARCHITECTURES=70
2822
ENV LD_LIBRARY_PATH=/opt/gluten/ep/build-velox/build/velox_ep/_build/release/_deps/curl-build/lib:$LD_LIBRARY_PATH
2923

30-
WORKDIR /opt/gluten
31-
RUN rm -rf /opt/rh/gcc-toolset-12 && ln -s /opt/rh/gcc-toolset-14 /opt/rh/gcc-toolset-12; \
24+
25+
RUN yum install -y sudo patch maven perl && \
3226
dnf remove -y cuda-toolkit-12* && dnf install -y cuda-toolkit-13-1; \
3327
dnf autoremove -y && dnf clean all; \
34-
source /opt/rh/gcc-toolset-12/enable; \
35-
bash ./dev/buildbundle-veloxbe.sh --run_setup_script=OFF --build_arrow=ON --spark_version=3.4 --build_tests=ON --build_benchmarks=ON --enable_gpu=ON && rm -rf /opt/gluten
28+
ln -sf /usr/local/bin/cmake /usr/bin && \
29+
git clone --depth=1 https://github.com/apache/incubator-gluten /opt/gluten && \
30+
cd /opt/gluten/.github/workflows/util/ && \
31+
./install-spark-resources.sh 3.4 && \
32+
cd /opt/gluten && \
33+
source /opt/rh/gcc-toolset-14/enable && \
34+
bash ./dev/buildbundle-veloxbe.sh --run_setup_script=OFF --build_arrow=ON --spark_version=3.4 --build_tests=ON --build_benchmarks=ON --enable_gpu=ON && \
35+
rm -rf /opt/gluten && \
36+
rm -rf /opt/spark-3.4.4-bin-hadoop3 && \
37+
rm -rf /root/.cache/ccache
38+
3639

3740
# You can try the data in folder backends-velox/src/test/resources/tpch-data-parquet
3841

0 commit comments

Comments
 (0)