silaseic commited on
Commit
1ee4f8e
1 Parent(s): e942976

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +6 -6
Dockerfile CHANGED
@@ -1,6 +1,6 @@
1
  # Start with a base image that includes CUDA
2
- # FROM nvidia/cuda:11.4.1-base-ubuntu20.04
3
- FROM huggingface/accelerate-gpu
4
 
5
  # Set the timezone
6
  ENV TZ=Europe/Berlin
@@ -25,10 +25,10 @@ RUN python3 -m pip install --no-cache-dir --upgrade -r /code/requirements.txt
25
  # Set up a new user named "user" with user ID 1000
26
  RUN useradd -m -u 1000 user
27
  # Add write permissions for others to the /usr/local/lib/python3.9/site-packages/oemer/checkpoints/seg_net/ directory
28
- # RUN chmod o+w /usr/local/lib/python3.10/dist-packages/oemer/checkpoints/seg_net/
29
- # RUN chmod o+w /usr/local/lib/python3.10/dist-packages/oemer/checkpoints/unet_big/
30
- RUN chmod o+w /opt/conda/lib/python3.10/site-packages/oemer/checkpoints/unet_big/
31
- RUN chmod o+w /opt/conda/lib/python3.10/site-packages/oemer/checkpoints/seg_net/
32
 
33
 
34
  RUN pip3 list -v
 
1
  # Start with a base image that includes CUDA
2
+ FROM nvidia/cuda:11.4.1-base-ubuntu20.04
3
+ # FROM huggingface/accelerate-gpu
4
 
5
  # Set the timezone
6
  ENV TZ=Europe/Berlin
 
25
  # Set up a new user named "user" with user ID 1000
26
  RUN useradd -m -u 1000 user
27
  # Add write permissions for others to the /usr/local/lib/python3.9/site-packages/oemer/checkpoints/seg_net/ directory
28
+ RUN chmod o+w /usr/local/lib/python3.10/dist-packages/oemer/checkpoints/seg_net/
29
+ RUN chmod o+w /usr/local/lib/python3.10/dist-packages/oemer/checkpoints/unet_big/
30
+ # RUN chmod o+w /opt/conda/lib/python3.10/site-packages/oemer/checkpoints/unet_big/
31
+ # RUN chmod o+w /opt/conda/lib/python3.10/site-packages/oemer/checkpoints/seg_net/
32
 
33
 
34
  RUN pip3 list -v