alvanli commited on
Commit
610c132
1 Parent(s): 9e1e2ca

Try runtime for docker image

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -1,4 +1,5 @@
1
- FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-devel as base
 
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