databricks / containers

Sample base images for Databricks Container Services
Apache License 2.0
167 stars 118 forks source link

[ML-33605] enabled cuda 11.8 for DCS image #129

Closed ygong1 closed 1 year ago

ygong1 commented 1 year ago

Enabled cuda 11.8 for DCS Image. Since DBR 9.x, conda is no longer supported. Use virtualenv to build the cuda dependency as well as the dependency need to start databricks REPL.

Tested