diff --git a/docker/torch1.13.0-cuda11.6-cudnn8.dockerfile b/docker/torch1.13.0-cuda11.6-cudnn8.dockerfile
index c85d3cf17..24930122f 100644
--- a/docker/torch1.13.0-cuda11.6-cudnn8.dockerfile
+++ b/docker/torch1.13.0-cuda11.6-cudnn8.dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:1.13.0-cuda11.6-cudnn8-runtime
+FROM pytorch/pytorch:1.13.0-cuda11.6-cudnn8-devel
 
 # Install dependencies
 RUN pip install --no-cache-dir \
@@ -13,7 +13,7 @@ RUN pip install --no-cache-dir \
       sentencepiece>=0.1.96 \
       tensorboard \
       typeguard \
-      dill \
+      dill
 
 RUN git clone https://github.com/k2-fsa/icefall /workspace/icefall && \
 	cd /workspace/icefall && \