add support for a100

2025-09-03 11:01:20 +08:00
parent f0425d03b4
commit fcf2001f24
3 changed files with 10 additions and 0 deletions
--- a/Dockerfile.a100
+++ b/Dockerfile.a100
@@ -0,0 +1,5 @@
+FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel
+
+WORKDIR /workspace
+RUN pip install sentence-transformers -i https://nexus.4pd.io/repository/pypi-all/simple
+COPY . /workspace/