sparkleman commited on
Commit
235bbc4
·
1 Parent(s): 50f89e3
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -25,7 +25,8 @@ RUN <<EOF
25
  apt update
26
  apt install --no-install-recommends -y \
27
  build-essential \
28
- git
 
29
  apt clean && rm -rf /var/lib/apt/lists/*
30
  EOF
31
 
 
25
  apt update
26
  apt install --no-install-recommends -y \
27
  build-essential \
28
+ git \
29
+ cuda-toolkit-12-4
30
  apt clean && rm -rf /var/lib/apt/lists/*
31
  EOF
32