JohnSnowLabs / spark-nlp

State of the Art Natural Language Processing
https://sparknlp.org/
Apache License 2.0
3.87k stars 711 forks source link

[SPARKNLP-1095] Add installation instructions for ONNX GPU on Databricks #14451

Open DevinTDHa opened 1 week ago

DevinTDHa commented 1 week ago

Description

This PR adds instructions on how to upgrade cuDNN on Databricks to infer ONNX models with GPU.

Additionally, the Databricks sections have been cleaned up a bit (removed redundant installation Instructions)