canonical / canonical/data-science-stack

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

Open
#36 1 comment 0 reactions 0 assignees View on GitHub
enhancement
Dominant language
Python
Stars
37
Forks
10
PR merge metrics
No merged PRs in 30d

Description

### 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](https://github.com/canonical/charmed-kubeflow-uats/tree/main) repository with instructions on how to run them.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.