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>
正在显示
请
注册
或者
登录
后发表评论