Dockerfile

# Dockerfile
FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime

# 安装依赖
RUN pip install medsam monai itk torchvision SimpleITK

# 复制模型权重
COPY medsam_vit_b.pth /app/medsam_vit_b.pth

# 复制应用代码
COPY app.py /app/app.py

# 设置启动命令
CMD python /app/app.py