Spaces:
Runtime error
Runtime error
Anooj
commited on
Commit
•
d4b38bf
1
Parent(s):
2203c22
try to build with gpus
Browse files- 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
|