diff --git a/INSTALL.md b/INSTALL.md
index abf3ef46a4552cb72a06b44ac459a632f01979f3..f2ca13e24fc3d6fbbb072ada9b9fca9ca647d911 100644
--- a/INSTALL.md
+++ b/INSTALL.md
@@ -153,9 +153,18 @@ $ docker run -it -p 19530:19530 -d milvusdb/milvus-cpu-build-env:latest
 
 Start a GPU container:
 
+- For nvidia docker 2:
+
 ```shell
 $ docker run --runtime=nvidia -it -p 19530:19530 -d milvusdb/milvus-gpu-build-env:latest
 ```
+
+- For nvidia container toolkit:
+
+```shell
+docker run --gpus all -it -p 19530:19530 -d milvusdb/milvus-gpu-build-env:latest
+```
+
 To enter the container:
 
 ```shell