From 1946db0d073799e1e43689122c025d052ff155db Mon Sep 17 00:00:00 2001
From: Stanislas0 <qinkai.zheng1028@gmail.com>
Date: Tue, 25 Apr 2023 11:15:54 +0800
Subject: [PATCH] Update README.md

---
 README.md | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/README.md b/README.md
index 47da7c6..d4746b1 100644
--- a/README.md
+++ b/README.md
@@ -73,6 +73,12 @@ git clone git@github.com:THUDM/CodeGeeX.git
 cd CodeGeeX
 pip install -e .
 ```
+Or use [CodeGeeX docker](https://hub.docker.com/r/codegeex/codegeex) to quickly set up the environment (with [nvidia-docker](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html#docker) installed):
+```bash
+docker pull codegeex/codegeex:latest
+# To enable GPU support, clarify device ids with --device
+docker run --gpus '"device=0,1"' -it --ipc=host --name=codegeex codegeex/codegeex
+```
 
 ### Model Weights