Skip to content

Commit ee3d3c7

Browse files
committed
PyTorch to 1.13.1+cu117
1 parent 9b916f3 commit ee3d3c7

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

modules/ObjectDetectionYOLOv5-6.2/requirements.linux.cuda11.txt

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -9,11 +9,11 @@ Pillow<10.0.0 # Installing Pillow, a Python Image Library
99
SciPy # Installing SciPy, a library for mathematics, science, and engineering
1010
PyYAML # Installing PyYAML, a library for reading configuration files
1111

12-
## For CUDA 11.7 (NOT torch 2.0+)
13-
--find-links https://download.pytorch.org/whl/torch_stable.html
14-
torch==1.13.0+cu117 # Installing PyTorch, an open source machine learning framework
15-
--find-links https://download.pytorch.org/whl/torch_stable.html
16-
torchvision==0.14.0+cu117 # Installing TorchVision, for working with computer vision models
12+
## For CUDA 11.7 (will work in CUDA 11.8)
13+
--extra-index-url https://download.pytorch.org/whl/cu117
14+
torch==1.13.1+cu117 # Installing PyTorch, an open source machine learning framework
15+
--extra-index-url https://download.pytorch.org/whl/cu117
16+
torchvision==0.14.1+cu117 # Installing TorchVision, for working with computer vision models
1717

1818
yolov5==6.2.3 # Installing Ultralytics YoloV5 package for object detection in images
1919

0 commit comments

Comments
 (0)