Update Dockerfile.tmpl

Author: djherbis

Dockerfile.tmpl

@@ -90,7 +90,7 @@ RUN conda config --add channels nvidia && \
 # b/232247930: uninstall pyarrow to avoid double installation with the GPU specific version.
 RUN pip uninstall -y pyarrow && \
     conda install -c conda-forge mamba && \
-    mamba install -y cudf cuml cudatoolkit && \
+    mamba install -y cudf cuml cudatoolkit==11.2.2 && \
     /tmp/clean-layer.sh
 {{ end }}