TensorFlow v2.18

OpenNMT · Feb 25, 2025 · 9464a13 · 9464a13
1 parent fc2ecf9
commit 9464a13
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/tests/requirements.txt b/python/tests/requirements.txt
@@ -2,7 +2,7 @@ transformers==4.41.*;platform_system=='Linux'
 fairseq==0.12.2;platform_system=='Linux' or platform_system=='Darwin'
 OpenNMT-py==2.2.*;platform_system=='Linux' or platform_system=='Darwin'
 OpenNMT-tf==2.30.*
-tensorflow-cpu==2.11.*
+tensorflow-cpu==2.18.*
 pytest
 wurlitzer==3.0.*;platform_system=='Linux'
 torch==2.2.0
diff --git a/tools/benchmark/opennmt_ende_wmt14/opennmt_tf/Dockerfile b/tools/benchmark/opennmt_ende_wmt14/opennmt_tf/Dockerfile
@@ -1,4 +1,4 @@
-FROM tensorflow/tensorflow:2.11.0-gpu
+FROM tensorflow/tensorflow:2.18.0-gpu
 
 RUN curl -O https://opennmt-models.s3.amazonaws.com/averaged-ende-ckpt500k-v2.tar.gz && \
     tar xf *.tar.gz && \