MVPilgrim commited on
Commit
9f74649
·
1 Parent(s): cde7628
Files changed (1) hide show
  1. Dockerfile +8 -0
Dockerfile CHANGED
@@ -13,6 +13,14 @@ RUN apt-get update && apt-get install -y \
13
  musl-tools \
14
  libffi-dev \
15
  git \
 
 
 
 
 
 
 
 
16
  && rm -rf /var/lib/apt/lists/*
17
 
18
  # Install Python 3.11
 
13
  musl-tools \
14
  libffi-dev \
15
  git \
16
+ build-essential \
17
+ cmake \
18
+ pkg-config \
19
+ cuda-toolkit-12-6 \
20
+ cuda-nvcc-12-6 \
21
+ libcublas-12-6 \
22
+ libcudnn8 \
23
+ && apt-get clean \
24
  && rm -rf /var/lib/apt/lists/*
25
 
26
  # Install Python 3.11