try cuda 11.3
Browse files- Dockerfile +2 -1
- README.md +5 -5
Dockerfile
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
-
FROM nvidia/cuda:11.
|
|
|
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 |
-
|
11 |
-
|
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
|