ariel-eddie commited on
Commit
079952f
·
verified ·
1 Parent(s): 7aa7542

Updated missing packages for YOLO

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -1
Dockerfile CHANGED
@@ -1,9 +1,11 @@
1
  # Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
2
  # you will also find guides on how best to write your Dockerfile
3
 
4
- FROM nvidia/cuda:12.5.1-cudnn-devel-ubuntu20.04
5
 
6
  RUN useradd -m -u 1000 user
 
 
7
  USER user
8
  ENV PATH="/home/user/.local/bin:$PATH"
9
 
 
1
  # Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
2
  # you will also find guides on how best to write your Dockerfile
3
 
4
+ FROM python:3.9
5
 
6
  RUN useradd -m -u 1000 user
7
+ RUN apt-get update && apt-get install ffmpeg libsm6 libxext6 -y
8
+
9
  USER user
10
  ENV PATH="/home/user/.local/bin:$PATH"
11