rapidsai / rapidsai/integration

Test Deep Learning Frameworks in nightlies

Open
#794 0 comments 1 reaction 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Shell
Stars
17
Forks
72
Avg merge
5d 5h
Merged PRs (30d)
3

Description

@caryr35 and I were discussing and we would like to add automated nightly tests that spot-check RAPIDS + Deep Learning Frameworks.

Currently there is a manual process for running smoke tests. It is mostly imports and ensuring GPU support is available.

We can automate this process with CI for our nightlies.

PyTorch

First, an environment is created:

conda create -n rapids-25.08 -c rapidsai -c conda-forge -c nvidia rapids=25.08 python=3.13 'cuda-version>=12.0,<=12.9' 'pytorch=*=*cuda*'

Then the following smoke tests are run:

import cudf
import cuml
import cugraph
import torch
assert torch.cuda.is_available()
TensorFlow

First, an environment is created:
(Note: TensorFlow doesn't support Python 3.13 at the time of writing)

conda create -n rapids-25.08 -c rapidsai -c conda-forge -c nvidia rapids=25.08 python=3.12 'cuda-version>=12.0,<=12.9' tensorflow

Then the following smoke tests are run (modified slightly to assert rather than print):

import cudf
import cuml
import cugraph
import tensorflow as tf
gpus = tf.config.list_physical_devices("GPU")
print(gpus)
assert len(gpus) > 1

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

The issue names no files or existing CI entry points. Start by locating the repository's nightly CI workflows and current smoke-test automation, then reproduce the documented conda environments. Done means nightly jobs create the PyTorch and TensorFlow environments, import cudf, cuml, cugraph, and the framework, and verify GPU availability with the stated assertions.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch, shell, tensorflow
Domain
ci-cd, machine-learning, testing
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
42/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.