docker sctg/roco-idefics3:0.0.6

Dockerfile

@@ -1,13 +1,17 @@
-# build with: docker build . --tag sctg/roco-idefics3:0.0.5 --tag sctg/roco-idefics3:latest --push
+# build with: docker build . --tag sctg/roco-idefics3:0.0.6 --tag sctg/roco-idefics3:latest --push
 # run with 
-# docker run --gpus all --user=42420:42420 -p 8080:8080 -e HF_TOKEN=hf_TOKEN -it sctg/roco-idefics3:0.0.5 bash -i /start.sh sleep infinity hf_TOKEN
+# 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
 # docker run --gpus all --user=42420:42420 -p 8080:8080 -it sctg/roco-idefics3:0.0.5 bash -i /start.sh python /learn.py hf_...
 FROM nvidia/cuda:11.6.1-devel-ubuntu20.04
-# FROM nvidia/cuda:11.0.3-devel-ubuntu20.04
-# RUN mkdir -p /workspace
+ARG NODE_MAJOR="20"
 RUN /usr/sbin/addgroup --gid 42420 ovh
 RUN /usr/sbin/useradd -u 42420 --gid 42420 -m -d /workspace -s /bin/bash ovh
-RUN apt update -y && apt-get install -y curl git git-lfs screen
+RUN apt update -y && apt-get install -y curl git git-lfs screen \
+      && mkdir -p /etc/apt/keyrings \
+      && curl -fsSL https://deb.nodesource.com/gpgkey/nodesource-repo.gpg.key | gpg --dearmor -o /etc/apt/keyrings/nodesource.gpg \
+      && echo "deb [signed-by=/etc/apt/keyrings/nodesource.gpg] https://deb.nodesource.com/node_$NODE_MAJOR.x nodistro main" | tee /etc/apt/sources.list.d/nodesource.list \
+      && apt-get update \
+      && apt-get install nodejs -y
 COPY --chmod=777 start.sh /start.sh
 COPY learn.py /learn.py
 # Mandatory to run the jobs in rootless mode