meg HF staff commited on
Commit
0ce31eb
·
verified ·
1 Parent(s): ceca75d

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -1,11 +1,11 @@
1
- FROM nvidia/cuda:12.2.0-devel-ubuntu22.04
2
  FROM python:3.9
 
3
 
4
  RUN useradd -m -u 1000 user
5
 
6
  ENV PATH=/opt/conda/bin:$PATH
7
 
8
- RUN pip install --no-cache-dir --upgrade -r requirements.txt
9
  RUN git clone -b energy_star_dev https://github.com/huggingface/optimum-benchmark.git /optimum-benchmark && cd optimum-benchmark && pip install -e .
10
 
11
  USER user
 
 
1
  FROM python:3.9
2
+ FROM nvidia/cuda:12.2.0-devel-ubuntu22.04
3
 
4
  RUN useradd -m -u 1000 user
5
 
6
  ENV PATH=/opt/conda/bin:$PATH
7
 
8
+ RUN pip install --no-cache-dir --upgrade -r /requirements.txt
9
  RUN git clone -b energy_star_dev https://github.com/huggingface/optimum-benchmark.git /optimum-benchmark && cd optimum-benchmark && pip install -e .
10
 
11
  USER user