praeclarumjj3 commited on
Commit
035df8d
1 Parent(s): 91f0bb6

Add cuda and python

Browse files
Files changed (1) hide show
  1. Dockerfile +24 -4
Dockerfile CHANGED
@@ -1,8 +1,27 @@
1
- FROM python:3.8.15
2
- FROM nvidia/cuda:11.4-cudnn8-runtime-ubuntu18.04
3
  CMD nvidia-smi
4
 
 
 
 
 
 
 
 
 
 
5
  RUN useradd -ms /bin/bash admin
 
 
 
 
 
 
 
 
 
 
 
6
 
7
  ENV WORKDIR=/code
8
  WORKDIR $WORKDIR
@@ -11,8 +30,11 @@ RUN chmod 755 $WORKDIR
11
 
12
  RUN apt-get update
13
  RUN apt-get install ffmpeg libsm6 libxext6 -y
 
 
14
 
15
  COPY requirements.txt $WORKDIR/requirements.txt
 
16
 
17
  RUN pip install gradio --no-cache-dir
18
  RUN pip install --no-cache-dir --upgrade -r $WORKDIR/requirements.txt
@@ -22,8 +44,6 @@ COPY . .
22
  RUN pwd
23
  RUN ls
24
 
25
- COPY oneformer $WORKDIR/oneformer
26
-
27
  RUN sh deform_setup.sh
28
 
29
  USER admin
 
1
+ FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu18.04
 
2
  CMD nvidia-smi
3
 
4
+ ENV DEBIAN_FRONTEND noninteractive
5
+ RUN apt-get update && apt-get install -y \
6
+ git \
7
+ make build-essential libssl-dev zlib1g-dev \
8
+ libbz2-dev libreadline-dev libsqlite3-dev wget curl llvm \
9
+ libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev \
10
+ ffmpeg libsm6 libxext6 cmake libgl1-mesa-glx \
11
+ && rm -rf /var/lib/apt/lists/*
12
+
13
  RUN useradd -ms /bin/bash admin
14
+ USER admin
15
+
16
+ ENV HOME=/home/user \
17
+ PATH=/home/user/.local/bin:$PATH
18
+
19
+ RUN curl https://pyenv.run | bash
20
+ ENV PATH=$HOME/.pyenv/shims:$HOME/.pyenv/bin:$PATH
21
+ RUN pyenv install 3.8.15 && \
22
+ pyenv global 3.8.15 && \
23
+ pyenv rehash && \
24
+ pip install --no-cache-dir --upgrade pip setuptools wheel
25
 
26
  ENV WORKDIR=/code
27
  WORKDIR $WORKDIR
 
30
 
31
  RUN apt-get update
32
  RUN apt-get install ffmpeg libsm6 libxext6 -y
33
+ RUN nvidia-smi
34
+
35
 
36
  COPY requirements.txt $WORKDIR/requirements.txt
37
+ COPY oneformer $WORKDIR/oneformer
38
 
39
  RUN pip install gradio --no-cache-dir
40
  RUN pip install --no-cache-dir --upgrade -r $WORKDIR/requirements.txt
 
44
  RUN pwd
45
  RUN ls
46
 
 
 
47
  RUN sh deform_setup.sh
48
 
49
  USER admin