none commited on
Commit
ebb4897
·
1 Parent(s): e310801

Update cuda

Browse files
Files changed (2) hide show
  1. Dockerfile +4 -4
  2. start.sh +3 -1
Dockerfile CHANGED
@@ -1,9 +1,9 @@
1
- # build with: docker build . --tag sctg/roco-idefics3:0.0.14 --tag sctg/roco-idefics3:latest --push
2
  # run with
3
- # docker run --gpus all --user=42420:42420 -p 7000-8000:7000-8000 -p 8080:8080 -e HF_TOKEN=hf_TOKEN -it sctg/roco-idefics3:0.0.14 bash -i /start.sh sleep infinity
4
- # docker run --gpus all --user=42420:42420 -p 7000-8000:7000-8000 -p 8080:8080 -it sctg/roco-idefics3:0.0.14 bash -i /start.sh python /learn.py hf_...
5
  FROM cloudflare/cloudflared:latest as cloudflared
6
- FROM nvidia/cuda:11.6.1-devel-ubuntu20.04
7
  ARG NODE_MAJOR="20"
8
  RUN /usr/sbin/addgroup --gid 42420 ovh
9
  RUN /usr/sbin/useradd -u 42420 --gid 42420 -m -d /workspace -s /bin/bash ovh
 
1
+ # build with: docker build . --tag sctg/roco-idefics3:0.0.15 --tag sctg/roco-idefics3:latest --push
2
  # run with
3
+ # docker run --gpus all --user=42420:42420 -p 7000-8000:7000-8000 -p 8080:8080 -e HF_TOKEN=hf_TOKEN -it sctg/roco-idefics3:0.0.15 bash -i /start.sh sleep infinity
4
+ # docker run --gpus all --user=42420:42420 -p 7000-8000:7000-8000 -p 8080:8080 -it sctg/roco-idefics3:0.0.15 bash -i /start.sh python /learn.py hf_...
5
  FROM cloudflare/cloudflared:latest as cloudflared
6
+ FROM nvidia/cuda:12.6.2-devel-ubuntu22.04
7
  ARG NODE_MAJOR="20"
8
  RUN /usr/sbin/addgroup --gid 42420 ovh
9
  RUN /usr/sbin/useradd -u 42420 --gid 42420 -m -d /workspace -s /bin/bash ovh
start.sh CHANGED
@@ -27,12 +27,14 @@ fi
27
  # Launch cloudflared tunnel
28
  # CFTOKEN.enc is the Cloudflared encrypted token
29
  # CFTOKEN.enc contained an encrypted string encrypted with the HF_TOKEN
 
 
30
  if [ -f /workspace/.config/CFTOKEN.enc ]; then
31
  echo "Decrypting CFTOKEN"
32
  CFTOKEN_ENC=$(cat /workspace/.config/CFTOKEN.enc)
33
  echo "$CFTOKEN_ENC" | openssl aes-256-cbc -a -d -md sha256 -pass pass:"$HF_TOKEN" > /workspace/.config/cftoken
34
  export CFTOKEN=$(cat /workspace/.config/cftoken)
35
- /usr/local/bin/cloudflared --pidfile /tmp/$name.pid --autoupdate-freq 24h0m0s tunnel run --token $CFTOKEN &
36
  fi
37
 
38
  git clone https://huggingface.co/eltorio/IDEFICS3_ROCOv2
 
27
  # Launch cloudflared tunnel
28
  # CFTOKEN.enc is the Cloudflared encrypted token
29
  # CFTOKEN.enc contained an encrypted string encrypted with the HF_TOKEN
30
+ # CFTOKEN.enc was created with:
31
+ # echo "the_cf_tunnel_token" | openssl aes-256-cbc -base64 -md sha256 -pass pass:"$HF_TOKEN" > CFTOKEN.enc
32
  if [ -f /workspace/.config/CFTOKEN.enc ]; then
33
  echo "Decrypting CFTOKEN"
34
  CFTOKEN_ENC=$(cat /workspace/.config/CFTOKEN.enc)
35
  echo "$CFTOKEN_ENC" | openssl aes-256-cbc -a -d -md sha256 -pass pass:"$HF_TOKEN" > /workspace/.config/cftoken
36
  export CFTOKEN=$(cat /workspace/.config/cftoken)
37
+ /usr/local/bin/cloudflared --pidfile /tmp/cf.pid --autoupdate-freq 24h0m0s tunnel run --token $CFTOKEN &
38
  fi
39
 
40
  git clone https://huggingface.co/eltorio/IDEFICS3_ROCOv2