Unverified 提交 ee91dc9b authored 作者: David A. Macey's avatar David A. Macey 提交者: GitHub

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: 's avatarDavid A. Macey <davidamacey@gmail.com> * Update requirements.txt Signed-off-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com> * Update requirements.txt Signed-off-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com> Signed-off-by: 's avatarDavid A. Macey <davidamacey@gmail.com> Signed-off-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
上级 55fbac93
......@@ -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
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论