Skip to content

Commit

Permalink
use base image with correct cuda version
Browse files Browse the repository at this point in the history
  • Loading branch information
maxnth committed Jan 28, 2024
1 parent 0871bdb commit 3b6391f
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions Dockerfile
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
FROM nvidia/cuda:12.3.1-runtime-ubuntu22.04
FROM nvidia/cuda:11.2.2-runtime-ubuntu20.04

ARG CALAMARI_URL=https://github.com/Calamari-OCR/calamari/archive/master.zip

Expand All @@ -14,4 +14,5 @@ RUN apt-get clean
RUN python3.7 -m pip --no-cache-dir install --upgrade pip
RUN python3.7 -m pip --no-cache-dir install ${CALAMARI_URL}
RUN apt-get purge -y --auto-remove build-essential
RUN rm -rf /var/lib/apt/lists/*
RUN rm -rf /var/lib/apt/lists/*
RUN export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.2/targets/x86_64-linux/lib

0 comments on commit 3b6391f

Please sign in to comment.