Create Dockerfile
Browse files- Dockerfile +14 -0
Dockerfile
ADDED
|
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Dockerfile
|
| 2 |
+
FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
|
| 3 |
+
|
| 4 |
+
# 安装依赖
|
| 5 |
+
RUN pip install medsam monai itk torchvision SimpleITK
|
| 6 |
+
|
| 7 |
+
# 复制模型权重
|
| 8 |
+
COPY medsam_vit_b.pth /app/medsam_vit_b.pth
|
| 9 |
+
|
| 10 |
+
# 复制应用代码
|
| 11 |
+
COPY app.py /app/app.py
|
| 12 |
+
|
| 13 |
+
# 设置启动命令
|
| 14 |
+
CMD python /app/app.py
|