diff --git a/paddlespeech/t2s/modules/tacotron2/attentions.py b/paddlespeech/t2s/modules/tacotron2/attentions.py
index 7a9e833bb..50eb498d6 100644
--- a/paddlespeech/t2s/modules/tacotron2/attentions.py
+++ b/paddlespeech/t2s/modules/tacotron2/attentions.py
@@ -174,7 +174,7 @@ class AttLoc(nn.Layer):
             att_prev = att_prev / enc_hs_len.unsqueeze(-1).astype(
                 att_prev.dtype)
         else:
-            att_prev.unsqueeze(-1) # only for static graph
+            att_prev.unsqueeze(-1)  # only for static graph
 
         # att_prev: (utt, frame) -> (utt, 1, 1, frame)
         # -> (utt, att_conv_chans, 1, frame)
diff --git a/tools/Dockerfile b/tools/Dockerfile
index b2f2b3125..dd4b22d2f 100644
--- a/tools/Dockerfile
+++ b/tools/Dockerfile
@@ -1,4 +1,4 @@
-FROM registry.baidubce.com/paddlepaddle/paddle:latest-dev-cuda11.2-cudnn8.2-trt8.0-gcc82
+FROM registry.baidubce.com/paddlepaddle/paddle:latest-dev-cuda12.3-cudnn9.0-trt8.6-gcc12.2
 RUN apt-get update -y
 RUN apt-get -y install libsndfile1 lsof
-RUN pip3.8 install pytest-runner
+RUN pip3.10 install pytest-runner