Hugging Face's logo

Spaces:
malleswar
/
docker-sub
Runtime error

App Files Community
malleswar commited on
Commit
5e3385f
·
verified ·
1 Parent(s): 62ad24c

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -1,4 +1,5 @@
1
  FROM python:3.10-slim
 
2
 
3
  # Set environment variables for cache directories and fixes
4
  ENV TRANSFORMERS_CACHE=/app/.cache/huggingface
 
1
  FROM python:3.10-slim
2
+ FROM nvidia/cuda:12.3.0-cudnn9-runtime-ubuntu22.04
3
 
4
  # Set environment variables for cache directories and fixes
5
  ENV TRANSFORMERS_CACHE=/app/.cache/huggingface