mac9087 commited on
Commit
619cd00
·
verified ·
1 Parent(s): 13b098e

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +6 -1
Dockerfile CHANGED
@@ -13,7 +13,9 @@ ENV PYTHONUNBUFFERED=1
13
  # Disable GPU detection
14
  ENV CUDA_VISIBLE_DEVICES=""
15
 
16
- # Install system dependencies
 
 
17
  RUN apt-get update && apt-get install -y \
18
  build-essential \
19
  libgl1-mesa-glx \
@@ -21,6 +23,9 @@ RUN apt-get update && apt-get install -y \
21
  libsm6 \
22
  libxext6 \
23
  libxrender-dev \
 
 
 
24
  git \
25
  && apt-get clean \
26
  && rm -rf /var/lib/apt/lists/*
 
13
  # Disable GPU detection
14
  ENV CUDA_VISIBLE_DEVICES=""
15
 
16
+ ENV ATTN_BACKEND="native"
17
+ ENV SPCONV_ALGO="native"
18
+
19
  RUN apt-get update && apt-get install -y \
20
  build-essential \
21
  libgl1-mesa-glx \
 
23
  libsm6 \
24
  libxext6 \
25
  libxrender-dev \
26
+ cmake \
27
+ libopenblas-dev \
28
+ liblapack-dev \
29
  git \
30
  && apt-get clean \
31
  && rm -rf /var/lib/apt/lists/*