none commited on
Commit
f0a6209
1 Parent(s): e2b04d6

roco-idefics3:0.0.7 (add jupyter_copilot)

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -4
Dockerfile CHANGED
@@ -1,7 +1,7 @@
1
- # build with: docker build . --tag sctg/roco-idefics3:0.0.6 --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.6 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.6 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,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