From 7c3d017c9037348b06ee6471789bde31c03065eb Mon Sep 17 00:00:00 2001 From: mmcwilliams Date: Thu, 14 Dec 2023 14:17:50 -0500 Subject: [PATCH] Upgrade useage of torch for training on 4090, older versions dont work. See why there are so many warnings --- requirements_yolov5.txt | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/requirements_yolov5.txt b/requirements_yolov5.txt index 3ed9be7..a85505b 100644 --- a/requirements_yolov5.txt +++ b/requirements_yolov5.txt @@ -66,10 +66,10 @@ tensorflow-estimator==2.14.0 tensorflow-io-gcs-filesystem==0.34.0 termcolor==2.3.0 thop==0.1.1.post2209072238 -torch==1.11.0 -torchaudio==0.11.0 -torchtext==0.12.0 -torchvision==0.12.0 +torch==1.13.0 +torchaudio==0.13.0 +torchtext==0.14.0 +torchvision==0.14.0 tqdm==4.66.1 typing_extensions==4.8.0 tzdata==2023.3