add < 9 for cuDNN spec - Githubissues

elixir-nx / xla

Pre-compiled XLA extension

Apache License 2.0

83 stars 21 forks source link

Closed zacksiri closed 3 weeks ago

zacksiri commented 3 weeks ago

We should specify that cuDNN needs to be < 9, 8.9+ may lead people to think that version 9 also works.

From what I saw here are relevant issues #86 #79 that I believe this PR will prevent.

jonatanklosko commented 3 weeks ago

Thanks!