Spaces:
Runtime error
Runtime error
Anooj
commited on
Commit
•
8a1bc30
1
Parent(s):
d4b38bf
fix
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
FROM julia:1.8 as julia-base
|
2 |
-
FROM nvidia/cuda
|
3 |
|
4 |
# ubuntu 20.04 is derived from debian buster
|
5 |
COPY --from=julia-base /usr/local/julia /usr/local/julia
|
|
|
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
|