artificialguybr commited on
Commit
563304d
·
verified ·
1 Parent(s): 9ac2f85

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -1
requirements.txt CHANGED
@@ -1,3 +1,5 @@
1
  opencv-python
2
  ffmpeg-python
3
- torch
 
 
 
1
  opencv-python
2
  ffmpeg-python
3
+ torch
4
+ nvidia-cuda-toolkit
5
+ nvidia cuda-python