File size: 358 Bytes
14dbfa0
0d17912
071876d
64ad8e8
4554def
3556cd8
0d17912
 
50fac44
0d17912
d9e9ff0
 
 
 
b90dbee
0d17912
2ff39f0
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
FROM python:3.11

WORKDIR /opt
RUN apt-get update \
    && apt-get install swig -y
RUN pip install "litrl[demo, torchrl]~=0.0.16"

COPY ./src /opt/src
COPY ./run.py /opt/run.py

RUN mkdir -p /.cache/torchrl \
    && mkdir -p /.cache/huggingface \
    && chmod -R 777 /.cache/torchrl \
    && chmod -R 777 /.cache/huggingface

# Runtime args
CMD python run.py