Skip to content

Commit 6cc34d8

Browse files
authored
Create Dockerfile
1 parent f1e2198 commit 6cc34d8

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

9_rlhf/wip/container/Dockerfile

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
FROM 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.0.0-gpu-py310-cu118-ubuntu20.04-sagemaker
2+
3+
COPY requirements.txt, main.py, utils.py /opt/ml/code/
4+
5+
RUN pip install -r requirements.txt

0 commit comments

Comments
 (0)