Update requirements.txt comment https://pytorch.org/get-started/locally/ (#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>
正在显示
... | @@ -9,7 +9,7 @@ Pillow>=7.1.2 | ... | @@ -9,7 +9,7 @@ Pillow>=7.1.2 |
PyYAML>=5.3.1 | PyYAML>=5.3.1 | ||
requests>=2.23.0 | requests>=2.23.0 | ||
scipy>=1.4.1 | scipy>=1.4.1 | ||
torch>=1.7.0 | torch>=1.7.0 # see https://pytorch.org/get-started/locally/ (recommended) | ||
torchvision>=0.8.1 | torchvision>=0.8.1 | ||
tqdm>=4.64.0 | tqdm>=4.64.0 | ||
# protobuf<=3.20.1 # https://github.com/ultralytics/yolov5/issues/8012 | # protobuf<=3.20.1 # https://github.com/ultralytics/yolov5/issues/8012 | ||
... | ... |
请
注册
或者
登录
后发表评论