Spaces:
Running
on
Zero
Running
on
Zero
update environment.yaml
Browse files- environment.yaml +2 -2
environment.yaml
CHANGED
|
@@ -4,12 +4,12 @@ channels:
|
|
| 4 |
- nvidia
|
| 5 |
dependencies:
|
| 6 |
- python=3.10
|
|
|
|
| 7 |
- pytorch=2.0.1
|
| 8 |
- torchvision=0.15.2
|
| 9 |
-
- pytorch-cuda=11.8
|
| 10 |
- pip
|
|
|
|
| 11 |
- pip:
|
| 12 |
-
- numpy==1.23.5
|
| 13 |
- pillow==9.5.0
|
| 14 |
- scipy==1.13.0
|
| 15 |
- scikit-image==0.21.0
|
|
|
|
| 4 |
- nvidia
|
| 5 |
dependencies:
|
| 6 |
- python=3.10
|
| 7 |
+
- cudatoolkit=11.8
|
| 8 |
- pytorch=2.0.1
|
| 9 |
- torchvision=0.15.2
|
|
|
|
| 10 |
- pip
|
| 11 |
+
- numpy==1.23.5
|
| 12 |
- pip:
|
|
|
|
| 13 |
- pillow==9.5.0
|
| 14 |
- scipy==1.13.0
|
| 15 |
- scikit-image==0.21.0
|