none
commited on
Commit
•
f0a6209
1
Parent(s):
e2b04d6
roco-idefics3:0.0.7 (add jupyter_copilot)
Browse files- Dockerfile +5 -4
Dockerfile
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
-
# build with: docker build . --tag sctg/roco-idefics3:0.0.
|
2 |
# run with
|
3 |
-
# docker run --gpus all --user=42420:42420 -p 8080:8080 -e HF_TOKEN=hf_TOKEN -it sctg/roco-idefics3:0.0.
|
4 |
-
# docker run --gpus all --user=42420:42420 -p 8080:8080 -it sctg/roco-idefics3:0.0.
|
5 |
FROM nvidia/cuda:11.6.1-devel-ubuntu20.04
|
6 |
ARG NODE_MAJOR="20"
|
7 |
RUN /usr/sbin/addgroup --gid 42420 ovh
|
@@ -27,7 +27,8 @@ RUN . /workspace/.miniconda3/bin/activate \
|
|
27 |
&& pip install -U git+https://github.com/huggingface/transformers.git\
|
28 |
&& pip install huggingface_hub[cli] accelerate datasets peft\
|
29 |
&& pip install -U Pillow \
|
30 |
-
&& pip install -U torchvision torchaudio
|
|
|
31 |
RUN . /workspace/.miniconda3/bin/activate && conda install -y jupyter
|
32 |
RUN rm -f /workspace/miniconda.sh
|
33 |
# Mandatory to run the jobs in rootless mode
|
|
|
1 |
+
# build with: docker build . --tag sctg/roco-idefics3:0.0.7 --tag sctg/roco-idefics3:latest --push
|
2 |
# run with
|
3 |
+
# docker run --gpus all --user=42420:42420 -p 8080:8080 -e HF_TOKEN=hf_TOKEN -it sctg/roco-idefics3:0.0.7 bash -i /start.sh sleep infinity hf_TOKEN
|
4 |
+
# docker run --gpus all --user=42420:42420 -p 8080:8080 -it sctg/roco-idefics3:0.0.7 bash -i /start.sh python /learn.py hf_...
|
5 |
FROM nvidia/cuda:11.6.1-devel-ubuntu20.04
|
6 |
ARG NODE_MAJOR="20"
|
7 |
RUN /usr/sbin/addgroup --gid 42420 ovh
|
|
|
27 |
&& pip install -U git+https://github.com/huggingface/transformers.git\
|
28 |
&& pip install huggingface_hub[cli] accelerate datasets peft\
|
29 |
&& pip install -U Pillow \
|
30 |
+
&& pip install -U torchvision torchaudio \
|
31 |
+
&& pip install jupyter_copilot
|
32 |
RUN . /workspace/.miniconda3/bin/activate && conda install -y jupyter
|
33 |
RUN rm -f /workspace/miniconda.sh
|
34 |
# Mandatory to run the jobs in rootless mode
|