Spaces:
Build error
Build error
Update Dockerfile
Browse files- Dockerfile +6 -6
Dockerfile
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
# Start with a base image that includes CUDA
|
2 |
-
|
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 |
-
|
29 |
-
|
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
|