Skip to content

Commit

Permalink
Update requirements.txt comment https://pytorch.org/get-started/locally/
Browse files Browse the repository at this point in the history
 (#9576)

* 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 <davidamacey@gmail.com>

* Update requirements.txt

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update requirements.txt

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: David A. Macey <davidamacey@gmail.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
  • Loading branch information
davidamacey and glenn-jocher committed Sep 25, 2022
1 parent 55fbac9 commit ee91dc9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ Pillow>=7.1.2
PyYAML>=5.3.1
requests>=2.23.0
scipy>=1.4.1
torch>=1.7.0
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
Expand Down

0 comments on commit ee91dc9

Please sign in to comment.