Hugging Face's logo

Spaces:
AnyaSchen
/
whisper-websocket
Paused

App Files Community
AnyaSchen commited on
Commit
eb4ba32
·
1 Parent(s): 20abd1e

try to use cuda

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -1,7 +1,7 @@
1
  # Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
2
  # you will also find guides on how best to write your Dockerfile
3
 
4
- FROM nvidia/cuda:12.1.0-cudnn8-runtime-ubuntu22.04
5
 
6
  # Install Python and system dependencies
7
  RUN apt-get update && apt-get install -y \
@@ -42,6 +42,7 @@ ENV PYTHONUNBUFFERED=1
42
  ENV CUDA_HOME=/usr/local/cuda
43
  ENV PATH=${CUDA_HOME}/bin:${PATH}
44
  ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
 
45
 
46
  # Expose the port the server runs on
47
  EXPOSE 7860
 
1
  # Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
2
  # you will also find guides on how best to write your Dockerfile
3
 
4
+ FROM nvidia/cuda:12.2.0-cudnn8-devel-ubuntu22.04
5
 
6
  # Install Python and system dependencies
7
  RUN apt-get update && apt-get install -y \
 
42
  ENV CUDA_HOME=/usr/local/cuda
43
  ENV PATH=${CUDA_HOME}/bin:${PATH}
44
  ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
45
+ ENV CUDNN_PATH=/usr/lib/x86_64-linux-gnu/libcudnn.so
46
 
47
  # Expose the port the server runs on
48
  EXPOSE 7860