Hugging Face's logo

Spaces:
atharvda
/
gemma-vision-judge
Paused

App Files Community
atharvda commited on
Commit
9f1eaf6
·
verified ·
1 Parent(s): 8dc839a

Upload Dockerfile with huggingface_hub

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- FROM pytorch/pytorch:2.5.1-cuda12.1-cudnn9-runtime
2
 
3
  ENV HF_HOME=/tmp/hf \
4
  HF_HUB_ENABLE_HF_TRANSFER=1 \
 
1
+ FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-runtime
2
 
3
  ENV HF_HOME=/tmp/hf \
4
  HF_HUB_ENABLE_HF_TRANSFER=1 \