Hugging Face's logo

Spaces:
WhisperTube
/
whispertube_backend
Runtime error

App Files Community
uzi007 commited on
Commit
aff20b4
•
1 Parent(s): 6101442

Updated Dockerfile for GPU Dependencies

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -1,5 +1,5 @@
1
  # Use an appropriate base image with Ubuntu
2
- FROM nvidia/cuda:11.2.1-base-ubuntu20.04
3
 
4
  # Set environment variables for CUDA
5
  ENV CUDA_HOME=/usr/local/cuda
 
1
  # Use an appropriate base image with Ubuntu
2
+ FROM nvidia/cuda:11.2.2-base-ubuntu20.04
3
 
4
  # Set environment variables for CUDA
5
  ENV CUDA_HOME=/usr/local/cuda