CodeProject.AI-ObjectDetectionYOLOv5-3.1/requirements.linux.cuda11.txt at main · codeproject/CodeProject.AI-ObjectDetectionYOLOv5-3.1 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
OpenCV-Python    # Installing OpenCV, the Open source Computer Vision library
Cython           # Installing the Cython compiler for C extensions for the Python language.
Pillow<10.0.0    # Installing Pillow, a Python Image Library
SciPy            # Installing SciPy, a library for mathematics, science, and engineering
PyYAML           # Installing PyYAML, a library for reading configuration files

tqdm             # Installing TDQM, the Fast, Extensible Progress Meter
Matplotlib       # Installing Matplotlib, the Python plotting package

## For CUDA 11.7
# AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor'
# https://github.com/ultralytics/yolov5/issues/6948
#--find-links https://download.pytorch.org/whl/torch_stable.html
#torch==1.13.0+cu117     # Installing PyTorch, an open source machine learning framework
#--find-links https://download.pytorch.org/whl/torch_stable.html
#torchvision==0.14.0+cu117 # Installing TorchVision, for working with computer vision models

torch                         # Installing PyTorch, an open source machine learning framework
torchvision                   # Installing TorchVision, for working with computer vision models

CodeProject-AI-SDK            # Installing the CodeProject.AI SDK

# Last line empty