Jhp commited on
Commit
5027026
·
1 Parent(s): 60d56ca
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -3,5 +3,5 @@ opencv-python
3
  wandb
4
  imageio
5
  scipy
6
- torch
7
- torchvision
 
3
  wandb
4
  imageio
5
  scipy
6
+ torch==1.11.0+cu113
7
+ torchvision==0.12.0+cu113