-
https://pytorch.org/get-started/locally/由 David A. Macey 提交于
* 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>
This project manages its dependencies using
pip.
Learn more