Forráskód Böngészése

Update Dockerfile to include comments for GPU architecture compatibility based on Compute Capability

myhloli 1 hónapja
szülő
commit
ef69f98cd6
2 módosított fájl, 13 hozzáadás és 2 törlés
  1. 8 1
      docker/china/Dockerfile
  2. 5 1
      docker/global/Dockerfile

+ 8 - 1
docker/china/Dockerfile

@@ -1,9 +1,16 @@
-# Use DaoCloud mirrored vllm image for China region
+# Use DaoCloud mirrored vllm image for China region for gpu with Ampere architecture and above (Compute Capability>=8.0)
+# Compute Capability version query (https://developer.nvidia.com/cuda-gpus)
 FROM docker.m.daocloud.io/vllm/vllm-openai:v0.10.1.1
 
 # Use the official vllm image
 # FROM vllm/vllm-openai:v0.10.1.1
 
+# Use DaoCloud mirrored vllm image for China region for gpu with Turing architecture and below (Compute Capability<8.0)
+# FROM docker.m.daocloud.io/vllm/vllm-openai:v0.10.2
+
+# Use the official vllm image
+# FROM vllm/vllm-openai:v0.10.2
+
 # Install libgl for opencv support & Noto fonts for Chinese characters
 RUN apt-get update && \
     apt-get install -y \

+ 5 - 1
docker/global/Dockerfile

@@ -1,6 +1,10 @@
-# Use the official vllm image
+# Use the official vllm image for gpu with Ampere architecture and above (Compute Capability>=8.0)
+# Compute Capability version query (https://developer.nvidia.com/cuda-gpus)
 FROM vllm/vllm-openai:v0.10.1.1
 
+# Use the official vllm image for gpu with Turing architecture and below (Compute Capability<8.0)
+# FROM vllm/vllm-openai:v0.10.2
+
 # Install libgl for opencv support & Noto fonts for Chinese characters
 RUN apt-get update && \
     apt-get install -y \