alvanli
commited on
Commit
•
610c132
1
Parent(s):
9e1e2ca
Try runtime for docker image
Browse files- Dockerfile +2 -1
Dockerfile
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
-
FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-
|
|
|
2 |
RUN apt-key adv --keyserver keyserver.ubuntu.com --recv-keys A4B469963BF863CC
|
3 |
RUN apt-get update && apt-get -y install git
|
4 |
|
|
|
1 |
+
FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime as base
|
2 |
+
|
3 |
RUN apt-key adv --keyserver keyserver.ubuntu.com --recv-keys A4B469963BF863CC
|
4 |
RUN apt-get update && apt-get -y install git
|
5 |
|