search

k2-fsa / k2

FSA/FST algorithms, differentiable, with PyTorch compatibility.
https://k2-fsa.github.io/k2
Apache License 2.0
1.1k stars 214 forks source link

Remove the constraint that k2 CUDA toolkit version has to match the torch CUDA toolkit version #1217

Closed yfyeung closed 1 year ago

yfyeung commented 1 year ago
The following combination has been tested. Pytorch Version Pytorch Cuda Version k2 Cuda Version
2.0.1 11.8 11.6
2.0.1 11.7 11.6
1.12.1 10.2 11.6