Update requirements.txt
Browse files- requirements.txt +1 -1
requirements.txt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
torch==1.11.0
|
| 2 |
torchvision==0.12.0
|
| 3 |
-
cudatoolkit
|
|
|
|
| 1 |
torch==1.11.0
|
| 2 |
torchvision==0.12.0
|
| 3 |
+
cudatoolkit==11.3.1
|