Spaces:
Runtime error
Runtime error
Commit ·
a39792d
1
Parent(s): aa154f2
fix: install CUDA-enabled torch for GPU spaces
Browse files- requirements.txt +1 -0
requirements.txt
CHANGED
|
@@ -2,6 +2,7 @@ openenv>=0.1.0
|
|
| 2 |
stable-baselines3>=2.3.0
|
| 3 |
sb3-contrib>=2.3.0
|
| 4 |
gymnasium>=0.29.1
|
|
|
|
| 5 |
torch>=2.2.0
|
| 6 |
numpy>=1.26.0
|
| 7 |
sentence-transformers>=3.0.0
|
|
|
|
| 2 |
stable-baselines3>=2.3.0
|
| 3 |
sb3-contrib>=2.3.0
|
| 4 |
gymnasium>=0.29.1
|
| 5 |
+
--extra-index-url https://download.pytorch.org/whl/cu121
|
| 6 |
torch>=2.2.0
|
| 7 |
numpy>=1.26.0
|
| 8 |
sentence-transformers>=3.0.0
|