 David A. MaceyandGlenn Jocher
|
ee91dc9bb3
|
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>
|
2022-09-25 14:47:16 +02:00 |
|