Anooj commited on
Commit
d4b38bf
1 Parent(s): 2203c22

try to build with gpus

Browse files
Files changed (1) hide show
  1. Dockerfile +14 -1
Dockerfile CHANGED
@@ -1,4 +1,17 @@
1
- FROM julia:1.8
 
 
 
 
 
 
 
 
 
 
 
 
 
2
 
3
  RUN useradd --create-home --shell /bin/bash genie
4
  RUN mkdir /home/genie/app
 
1
+ FROM julia:1.8 as julia-base
2
+ FROM nvidia/cuda:$11.2.0-cudnn8-devel-ubuntu20.04 as base
3
+
4
+ # ubuntu 20.04 is derived from debian buster
5
+ COPY --from=julia-base /usr/local/julia /usr/local/julia
6
+
7
+ # since we are copying from julia-base, need to add to PATH
8
+ ENV JULIA_PATH /usr/local/julia
9
+ ENV PATH $JULIA_PATH/bin:$PATH
10
+ # cuda env vars to use cuda & julia https://cuda.juliagpu.org/stable/installation/overview/
11
+ ENV JULIA_CUDA_USE_BINARYBUILDER="false"
12
+ ENV JULIA_DEBUG CUDA
13
+ ENV CUDA_HOME /usr/local/cuda
14
+
15
 
16
  RUN useradd --create-home --shell /bin/bash genie
17
  RUN mkdir /home/genie/app