Spaces:
Sleeping
Sleeping
Dockerfile updated
Browse files- Dockerfile +1 -1
- requirements.txt +6 -6
Dockerfile
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
FROM pytorch/pytorch:
|
2 |
|
3 |
WORKDIR /code
|
4 |
|
|
|
1 |
+
FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
|
2 |
|
3 |
WORKDIR /code
|
4 |
|
requirements.txt
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
-
torch==
|
2 |
-
torchvision==0.
|
3 |
gradio==3.35.2
|
4 |
-
Pillow==9.
|
5 |
-
numpy==1.
|
6 |
-
matplotlib==3.
|
7 |
-
scikit-learn==1.
|
|
|
1 |
+
torch==2.0.1
|
2 |
+
torchvision==0.15.2
|
3 |
gradio==3.35.2
|
4 |
+
Pillow==9.5.0
|
5 |
+
numpy==1.24.3
|
6 |
+
matplotlib==3.7.1
|
7 |
+
scikit-learn==1.2.2
|