developer-lunark commited on
Commit
f2119d1
·
verified ·
1 Parent(s): 30cc679

Use lighter base image and fix memory issue

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime
2
 
3
  WORKDIR /app
4
 
 
1
+ FROM python:3.11-slim
2
 
3
  WORKDIR /app
4