SerdarHelli commited on
Commit
c38bb0e
·
1 Parent(s): 4c5f3d0

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -1
requirements.txt CHANGED
@@ -4,4 +4,6 @@ tqdm
4
  imageio
5
  torch
6
  torchvision
7
- torch-utils
 
 
 
4
  imageio
5
  torch
6
  torchvision
7
+ torch-utils
8
+ pytorch=1.9.1
9
+ cudatoolkit=11.1