Nirav-Madhani commited on
Commit
f16b587
·
verified ·
1 Parent(s): a61e149

Create Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +11 -0
Dockerfile ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM python:3.10
2
+
3
+ WORKDIR /app
4
+ RUN git clone https://github.com/octo-models/octo.git
5
+ WORKDIR /app/octo
6
+ RUN pip3 install -e .
7
+ RUN pip3 install -r requirements.txt
8
+ RUN pip3 install --upgrade "jax[cuda11_pip]==0.4.20" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
9
+ RUN pip install scipy==1.10.1
10
+ COPY main.py /app/octo
11
+ CMD ["python", "main.py"]