acharyaaditya26 commited on
Commit
b133312
·
1 Parent(s): fb09930
Files changed (1) hide show
  1. Dockerfile +1 -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.3.2-cudnn9-devel-ubuntu22.04
5
 
6
  RUN apt-get update && apt-get upgrade -y
7
  RUN apt-get install -y python3-pip python3-dev
 
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.3.2-cudnn9-runtime-ubuntu22.04
5
 
6
  RUN apt-get update && apt-get upgrade -y
7
  RUN apt-get install -y python3-pip python3-dev