resolves #2542 modify dockerfile to upgrade cuda to 12.2.0 and pydant…

…ic 1.10.13 (#2565)
lm-sys · Oct 15, 2023 · cb71875 · cb71875
1 parent 9db2143
commit cb71875
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -1,7 +1,7 @@
-FROM nvidia/cuda:11.7.1-runtime-ubuntu20.04
+FROM nvidia/cuda:12.2.0-runtime-ubuntu20.04
 
 RUN apt-get update -y && apt-get install -y python3.9 python3.9-distutils curl
 RUN curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py
 RUN python3.9 get-pip.py
 RUN pip3 install fschat
-RUN pip3 install fschat[model_worker,webui] pydantic==1.10.1
+RUN pip3 install fschat[model_worker,webui] pydantic==1.10.13