From 320b8ec5c6888f2662b240d7d694ccd26ffc2640 Mon Sep 17 00:00:00 2001 From: "David A. Macey" Date: Sun, 25 Sep 2022 06:45:39 -0400 Subject: [PATCH 1/3] Update Requirements with PyTorch CUDA Added --extra-index-url https://download.pytorch.org/whl/cu116 URL to requirements file for ease of creating venv with CUDA enabled PyTorch. Otherwise CPU PyTorch is installed an unable to use local GPUs. Signed-off-by: David A. Macey --- requirements.txt | 2 ++ 1 file changed, 2 insertions(+) diff --git a/requirements.txt b/requirements.txt index 4d6ec3509efa..0fc82c7f4271 100644 --- a/requirements.txt +++ b/requirements.txt @@ -10,7 +10,9 @@ PyYAML>=5.3.1 requests>=2.23.0 scipy>=1.4.1 torch>=1.7.0 +--extra-index-url https://download.pytorch.org/whl/cu116 torchvision>=0.8.1 +--extra-index-url https://download.pytorch.org/whl/cu116 tqdm>=4.64.0 # protobuf<=3.20.1 # https://github.com/ultralytics/yolov5/issues/8012 From 79668a9233270c3f6cb562d7791a789e51986c85 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 25 Sep 2022 14:46:20 +0200 Subject: [PATCH 2/3] Update requirements.txt Signed-off-by: Glenn Jocher --- requirements.txt | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/requirements.txt b/requirements.txt index 0fc82c7f4271..09563f0606d5 100644 --- a/requirements.txt +++ b/requirements.txt @@ -9,10 +9,8 @@ Pillow>=7.1.2 PyYAML>=5.3.1 requests>=2.23.0 scipy>=1.4.1 -torch>=1.7.0 ---extra-index-url https://download.pytorch.org/whl/cu116 +torch>=1.7.0 # See https://pytorch.org/get-started/locally/ (recommended) torchvision>=0.8.1 ---extra-index-url https://download.pytorch.org/whl/cu116 tqdm>=4.64.0 # protobuf<=3.20.1 # https://github.com/ultralytics/yolov5/issues/8012 From 767a97cc9d23decbe7c21777007545dd28f80fcd Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 25 Sep 2022 14:46:40 +0200 Subject: [PATCH 3/3] Update requirements.txt Signed-off-by: Glenn Jocher --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 09563f0606d5..0436f415c642 100644 --- a/requirements.txt +++ b/requirements.txt @@ -9,7 +9,7 @@ Pillow>=7.1.2 PyYAML>=5.3.1 requests>=2.23.0 scipy>=1.4.1 -torch>=1.7.0 # See https://pytorch.org/get-started/locally/ (recommended) +torch>=1.7.0 # see https://pytorch.org/get-started/locally/ (recommended) torchvision>=0.8.1 tqdm>=4.64.0 # protobuf<=3.20.1 # https://github.com/ultralytics/yolov5/issues/8012