moriire commited on
Commit
421d43d
·
verified ·
1 Parent(s): d8f7c9d

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -13,7 +13,8 @@ RUN apt-get update && apt-get upgrade -y && apt-get install -y --no-install-reco
13
  python3-pip \
14
  ninja-build \
15
  libopenblas-dev \
16
- build-essential
 
17
 
18
  RUN mkdir /app
19
  WORKDIR /app
 
13
  python3-pip \
14
  ninja-build \
15
  libopenblas-dev \
16
+ build-essential \
17
+ install cuda-11-8
18
 
19
  RUN mkdir /app
20
  WORKDIR /app