-
Notifications
You must be signed in to change notification settings - Fork 2
/
requirements.linux.cuda12.txt
35 lines (27 loc) · 1.75 KB
/
requirements.linux.cuda12.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
#! Python3.8
# CUDA requirements for Linux with CUDA 12 installed
numpy<2.0 # Installing NumPy, a package for scientific computing
# Pandas # Installing Pandas, a data analysis / data manipulation tool
# CoreMLTools # Installing CoreMLTools, for working with .mlmodel format models
# OpenCV-Python # Installing OpenCV, the Open source Computer Vision library
# 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
## For CUDA 12.1 (nightly)
# --pre
# --index-url https://download.pytorch.org/whl/nightly/cu121
# torch # Installing PyTorch, an open source machine learning framework
# --pre
# --index-url https://download.pytorch.org/whl/nightly/cu121
# torchvision # Installing TorchVision, for working with computer vision models
## For CUDA 11.7 (will work in CUDA 12)
--extra-index-url https://download.pytorch.org/whl/cu117
torch==1.13.1+cu117 # Installing PyTorch, an open source machine learning framework
--extra-index-url https://download.pytorch.org/whl/cu117
torchvision==0.14.1+cu117 # Installing TorchVision, for working with computer vision models
# Specific version because we have a patch
ultralytics==8.1.2 # Installing Ultralytics package for object detection in images
# We need this, but we don't need this.
# Seaborn # Installing Seaborn, a data visualization library based on matplotlib
CodeProject-AI-SDK # Installing the CodeProject.AI SDK
# last line empty.