为长城机器学习项目创建的 sys 包
归 The Farlander 所有。
5 在 #cudnn
4KB
这是一个用于长城机器学习框架的 sys 包,使用 bindgen 创建了对 Cuda 和 Cudnn 的绑定。
Canyon-sys 需要查找环境变量 CUDNN_ROOT 和 CUDA_ROOT 才能正常运行。导出这些变量,然后构建 canyon-sys。
CUDNN_ROOT
CUDA_ROOT
~0–1.8MB ~36K SLoC
build.rs