pin to use appropriate cuda version

[Archana-Shinde1]

Checklist before submitting

• Did you read the contributor guide?
• Did you update any affected documentation?
• Did you write any tests to validate this change?

Description

Fixes # (issue).

Review process to land

1. All tests and other checks must succeed.
2. At least one maintainer must review and approve.
3. If any maintainer requests changes, they must be addressed.

[cdeepali]

I think we need the change in the run section too.

- cudatoolkit {{ cudatoolkit }} #[build_type == 'cuda']