Hugging Face's logo

Spaces:
jxtan
/
translator_api
Runtime error

App Files Community
jxtan commited on
Commit
9d08349
·
1 Parent(s): 76e433a

Updated base image

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -2
Dockerfile CHANGED
@@ -1,5 +1,4 @@
1
- FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04
2
- ENV DEBIAN_FRONTEND=noninteractive
3
 
4
  RUN useradd -m -u 1000 user
5
  USER user
 
1
+ FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-runtime
 
2
 
3
  RUN useradd -m -u 1000 user
4
  USER user