WIP: ensure 'torch' CUDA wheels are installed in CI, remove unused dependencies

[jameslamb]

Contributes to rapidsai/build-planning#257

• ensures that when torch is installed in wheels CI, it's always a CUDA variant
• removes unused test dependencies ogb, pydantic, torchdata, and torchmetrics
  • I suspect these are left over from before GNN packages moved out to https://github.com/rapidsai/cugraph-gnn

Notes for Reviewers

How does this help with testing against a mix of CTKs?

torch CUDA wheels tightly pin to nvidia-{thing} wheels and soon, cuda-toolkit (see rapidsai/build-planning#255).

Forcing the use of CUDA torch wheels ensures we'll catch dependency conflicts in CI. Without this, pip could silently fall back to CPU-only torch from pypi.org.

How did you test these changes?

I've been thoroughly testing these exact script + dependencies.yaml changes using RMM's PyTorch tests in rapidsai/rmm#2270

[copy-pr-bot]

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

[jameslamb]

/ok to test