search

canonical / data-science-stack

Stack with machine learning tools needed for local development.
Apache License 2.0
9 stars 3 forks source link

Create UATs that test `dss` actually uses GPUs (NVIDIA) #36

Open DnPlas opened 4 months ago

DnPlas commented 4 months ago

Why it needs to get done

In order to have e2e testing, an automated and repeatable testing framework is required to ensure dss can actually spin up Jupyter servers where GPU workloads can run on. Testing also covers the ML frameworks (pytorch and tensorflow) that users have access to, so this should also be considered.

What needs to get done

  1. Create notebooks that exercise Pytorch and Tensorflow in a CPU environment
  2. Create notebooks that exercise Pytorch and Tensorflow in a GPU environment

When is the task considered done

When the notebooks are placed in the UATs repository with instructions on how to run them.

syncronize-issues-to-jira[bot] commented 4 months ago

Thank you for reporting us your feedback!

The internal ticket has been created: https://warthogs.atlassian.net/browse/KF-5385.

This message was autogenerated