File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change 33# GPU: docker build -f Dockerfile.pb --build-arg BUILD_TYPE=gpu -t ramp-fair:gpu .
44
55ARG PY_VER=3.10
6- ARG TF_VER=2.9.2
6+ ARG TF_VER=2.9.3
77ARG BUILD_TYPE=gpu
88ARG CUDA_TAG=11.8.0-cudnn8-runtime-ubuntu22.04
9- ARG CUDA_DEVEL_TAG=11.8.0-cudnn8-devel-ubuntu22.04
9+ # ARG CUDA_DEVEL_TAG=11.8.0-cudnn8-devel-ubuntu22.04
1010
1111# ==============================================================================
1212# === CPU base image (minimal) =================================================
@@ -29,7 +29,7 @@ RUN apt-get update && \
2929 && rm -rf /var/lib/apt/lists/* && \
3030 python3 -m pip install --upgrade pip
3131
32- FROM nvidia/cuda:${CUDA_DEVEL_TAG} AS cuda-dev
32+ # FROM nvidia/cuda:${CUDA_DEVEL_TAG} AS cuda-dev
3333
3434
3535# ==============================================================================
@@ -76,7 +76,7 @@ ENV CPLUS_INCLUDE_PATH=/usr/include/gdal \
7676COPY --from=builder /usr/local /usr/local
7777COPY --from=builder /usr/lib/python*/ /usr/lib/python*/
7878COPY --from=builder /usr/include/gdal /usr/include/gdal
79- COPY --from=cuda-dev /usr/local/cuda/nvvm/libdevice /usr/local/cuda/nvvm/libdevice
79+ # COPY --from=cuda-dev /usr/local/cuda/nvvm/libdevice /usr/local/cuda/nvvm/libdevice
8080
8181WORKDIR /app
8282CMD ["bash"]
You can’t perform that action at this time.
0 commit comments