Ywung commited on
Commit
d9a9b99
1 Parent(s): b2650c5

try cuda 11.3

Browse files
Files changed (2) hide show
  1. Dockerfile +2 -1
  2. README.md +5 -5
Dockerfile CHANGED
@@ -1,4 +1,5 @@
1
- FROM nvidia/cuda:11.8.0-devel-ubuntu22.04 as builder
 
2
 
3
  RUN apt-get update && \
4
  apt-get install --no-install-recommends -y git vim build-essential python3-dev python3-venv && \
 
1
+ FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu18.04 as builder
2
+ # FROM nvidia/cuda:11.8.0-devel-ubuntu22.04 as builder
3
 
4
  RUN apt-get update && \
5
  apt-get install --no-install-recommends -y git vim build-essential python3-dev python3-venv && \
README.md CHANGED
@@ -4,11 +4,11 @@ emoji: 🤖
4
  colorFrom: blue
5
  colorTo: purple
6
  pinned: true
7
- sdk: gradio
8
- sdk_version: 3.41.2
9
- app_file: server.py
10
- # sdk: docker
11
- # app_port: 7860
12
  ---
13
 
14
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
4
  colorFrom: blue
5
  colorTo: purple
6
  pinned: true
7
+ # sdk: gradio
8
+ # sdk_version: 3.41.2
9
+ # app_file: server.py
10
+ sdk: docker
11
+ app_port: 7860
12
  ---
13
 
14
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference