diff --git a/docker/inference/dockerfile.torch b/docker/inference/dockerfile.torch
index 79e898076..5c4c9ea66 100644
--- a/docker/inference/dockerfile.torch
+++ b/docker/inference/dockerfile.torch
@@ -53,7 +53,7 @@ RUN pip install cupy-cuda115 nvidia-pyindex pybind11 pytest protobuf transformer
 RUN pip install betterproto cachetools graphviz nvtx scipy sklearn
 RUN pip install numba --no-deps
 RUN pip install tritonclient[all] grpcio-channelz
-RUN pip install dask==2021.11.2 distributed==2021.11.2 dask[dataframe]==2021.11.2 dask-cuda22.2.0
+RUN pip install dask==2021.11.2 distributed==2021.11.2 dask[dataframe]==2021.11.2 dask-cuda==22.2.0
 RUN pip install git+https://github.com/rapidsai/asvdb.git@main
 RUN pip install "cuda-python>=11.5,<12.0"