matthoffner commited on
Commit
2f314f4
·
verified ·
1 Parent(s): 761b350

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -7,6 +7,7 @@ ENV DEBIAN_FRONTEND=noninteractive
7
  RUN apt update && \
8
  apt install --no-install-recommends -y \
9
  build-essential \
 
10
  python3 \
11
  python3-pip \
12
  wget \
 
7
  RUN apt update && \
8
  apt install --no-install-recommends -y \
9
  build-essential \
10
+ nvidia-cuda-toolkit \
11
  python3 \
12
  python3-pip \
13
  wget \