matthoffner commited on
Commit
27a727f
·
1 Parent(s): c4010d9

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -2
Dockerfile CHANGED
@@ -1,6 +1,9 @@
1
- FROM python:latest
2
 
3
- ENV PYTHONUNBUFFERED 1
 
 
 
4
 
5
  EXPOSE 7860
6
 
 
1
+ FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu20.04
2
 
3
+ ARG DEBIAN_FRONTEND=noninteractive
4
+
5
+ RUN apt update
6
+ RUN apt install -y git python3 python3-pip
7
 
8
  EXPOSE 7860
9