@ -11,6 +11,7 @@ documentation = "https://docs.comma.ai"
[ tool . poetry . dependencies ]
python = "~3.11"
poetry = "~1.5.1"
atomicwrites = "*"
aiohttp = "*"
aiortc = "*"
@ -31,7 +32,6 @@ numpy = "==1.23.0" # locked pending deprecation fixes in xx
onnx = ">=1.14.0"
onnxruntime-gpu = { version = ">=1.15.1" , platform = "linux" , markers = "platform_machine == 'x86_64'" }
pillow = "*"
poetry = "==1.2.2"
protobuf = "==3.20.3"
psutil = "*"
pyaudio = "*"
@ -46,7 +46,7 @@ PyYAML = "*"
pyzmq = "*"
requests = "*"
scons = "*"
sentry-sdk = "* "
sentry-sdk = "~1.28 "
setproctitle = "*"
smbus2 = "*"
sounddevice = "*"
@ -131,6 +131,7 @@ datadog = "*"
dotmap = "*"
einops = "*"
elasticsearch = "*"
flake8 = "*"
Flask-Cors = "*"
Flask-SocketIO = "*"
GeoAlchemy2 = "*"
@ -155,6 +156,7 @@ pipenv = "==2022.10.12"
plotly = "*"
pycuda = "*"
Pygments = "*"
pylint = "*"
PyMySQL = "~0.9"
pyproj = "*"
python-logstash = "*"
@ -166,6 +168,7 @@ scikit-learn = "*"
segmentation-models-pytorch = "==0.3.3"
simplejson = "*"
SQLAlchemy = "*"
wrapt = "*"
torch = { url = "https://download.pytorch.org/whl/cu118/torch-2.0.1%2Bcu118-cp311-cp311-linux_x86_64.whl" }
torchsummary = "*"
torchvision = { url = "https://download.pytorch.org/whl/cu118/torchvision-0.15.2%2Bcu118-cp311-cp311-linux_x86_64.whl" }
@ -177,7 +180,7 @@ osmnx = "==1.2.2"
tritonclient = { version = "2.28.0" , extras = [ "http" ] }
transformers = "*"
timm = "==0.9.2"
PyNvCodec = { git = "https://github.com/NVIDIA/VideoProcessingFramework.git" , rev = "3347e55" }
PyNvCodec = { git = "https://github.com/NVIDIA/VideoProcessingFramework.git" }
apex = { url = "https://github.com/commaai/apex/releases/download/pytorch2.0.1%2Bcu11.8/apex-0.1-cp311-cp311-linux_x86_64.whl" }
opencv-python-headless = { url = "https://github.com/commaai/opencv-python-builder/releases/download/4.5.5.64%2Bcu118-cp311/opencv_python_headless-4.5.5.64-cp311-cp311-manylinux_2_31_x86_64.whl" , platform = "linux" }