mattmcw
/
yolo_web
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Upgrade useage of torch for training on 4090, older versions dont work. See why there are so many warnings

This commit is contained in:
mmcwilliams 2023-12-14 14:17:50 -05:00
parent a560b9fdab
commit 7c3d017c90
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
requirements_yolov5.txt
View File

@ -66,10 +66,10 @@ tensorflow-estimator==2.14.0
tensorflow-io-gcs-filesystem==0.34.0 tensorflow-io-gcs-filesystem==0.34.0
termcolor==2.3.0 termcolor==2.3.0
thop==0.1.1.post2209072238 thop==0.1.1.post2209072238
torch==1.11.0 torch==1.13.0
torchaudio==0.11.0 torchaudio==0.13.0
torchtext==0.12.0 torchtext==0.14.0
torchvision==0.12.0 torchvision==0.14.0
tqdm==4.66.1 tqdm==4.66.1
typing_extensions==4.8.0 typing_extensions==4.8.0
tzdata==2023.3 tzdata==2023.3