AnyaSchen commited on
Commit
070d9af
·
1 Parent(s): eb4ba32

try to use cuda

Browse files
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.2.0-cudnn8-devel-ubuntu22.04
5
 
6
  # Install Python and system dependencies
7
  RUN apt-get update && apt-get install -y \
 
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.8.1-cudnn-devel-ubuntu22.04
5
 
6
  # Install Python and system dependencies
7
  RUN apt-get update && apt-get install -y \