cuda-cudnn-sys

cuDNN的Rust封装，为autograph创建 https://github.com/charles-r-earp/autograph

要求

必须安装CUDA https://developer.nvidia.com/cuda-downloads 和 cuDNN https://developer.nvidia.com/cudnn。有关其他帮助，请参阅 https://github.com/bheisler/RustaCUDA。如果您在Unix平台上未在 "/usr/local/cuda/include" 中找到 cuda.h，则需要设置环境变量 CUDA_INCLUDE_DIR。此变量必须在Windows上设置。

支持的平台

在Ubuntu-18.04上进行了测试。应能在Linux、macOS和Windows上运行。如果您遇到构建错误，请创建一个问题。

测试

通过运行以下命令来验证crate是否工作

cargo test