deepjavalibrary / djl-serving

A universal scalable machine learning model deployment solution
Apache License 2.0
199 stars 67 forks source link

[sm][docker] use the correct verified cuda version in docker file #2565

Closed siddvenk closed 6 days ago

siddvenk commented 6 days ago

Description

hosting has a strict cuda version check on the label, it needs to 12.2 to use the proper driver. this was fixed in 0.30.0, but seems like we missed it on the main branch