Skip to content

Commit 84663e3

Browse files
committed
[docker] Upgrade to cuda 12.9
1 parent a8f947a commit 84663e3

File tree

2 files changed

+3
-7
lines changed

2 files changed

+3
-7
lines changed

serving/docker/cuda-compat

Lines changed: 0 additions & 3 deletions
This file was deleted.

serving/docker/lmi.Dockerfile

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -9,9 +9,9 @@
99
# or in the "LICENSE.txt" file accompanying this file. This file is distributed on an "AS IS"
1010
# BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, express or implied. See the License for
1111
# the specific language governing permissions and limitations under the License.
12-
ARG version=12.8.1-devel-ubuntu24.04
12+
ARG version=12.9.1-devel-ubuntu24.04
1313
FROM nvidia/cuda:$version
14-
ARG cuda_version=cu128
14+
ARG cuda_version=cu129
1515
ARG djl_version
1616
ARG djl_serving_version
1717
ARG python_version=3.12
@@ -61,7 +61,6 @@ ENV PATH="/opt/djl/bin:${PATH}"
6161
ENTRYPOINT ["/usr/local/bin/dockerd-entrypoint.sh"]
6262
CMD ["serve"]
6363

64-
COPY cuda-compat /etc/apt/preferences.d/cuda-compat
6564
COPY scripts scripts/
6665
RUN chmod -R +x scripts
6766
RUN mkdir -p /opt/djl/conf \
@@ -74,7 +73,7 @@ COPY config.properties /opt/djl/conf/config.properties
7473
COPY partition /opt/djl/partition
7574
COPY scripts/telemetry.sh /opt/djl/bin
7675

77-
RUN apt-get update && apt-get install -yq libaio-dev libopenmpi-dev g++ unzip cuda-compat-12-8 \
76+
RUN apt-get update && apt-get install -yq libaio-dev libopenmpi-dev g++ unzip cuda-compat-12-9 \
7877
&& scripts/install_openssh.sh \
7978
&& scripts/install_python.sh ${python_version} \
8079
&& scripts/install_s5cmd.sh x64 \

0 commit comments

Comments
 (0)