2个不稳定版本
使用旧Rust 2015
0.2.0 | 2016年2月4日 |
---|---|
0.1.0 | 2016年1月18日 |
在 机器学习 中排名第 719
每月下载量 24 次
在 2 个crates中使用(通过 collenchyma-blas)
41KB
763 行
rust-cuBLAS •
rust-cuBLAS为CUDA的cuBLAS库提供了一个安全包装器,这样您就可以在Rust应用程序中舒适且安全地使用cuBLAS。
由于cuBLAS目前依赖于CUDA在GPU上分配内存,您可能还需要查看rust-cuda。
rust-cublas是在[Autumn][autumn]为Rust机器智能框架Leaf开发的。
rust-cublas是高性能计算框架Collenchyma的一部分,用于BLAS插件。对于BLAS操作(如cuBLAS提供的)的简单、统一接口,您可能需要查看Collenchyma。
有关更多信息,
入门指南
如果您使用Cargo,只需将rust-cuBLAS添加到您的Cargo.toml
[dependencies]
cublas = "0.2.0"
如果您使用Cargo Edit,可以调用
$ cargo add cublas
构建
rust-cublas 依赖于 cuBLAS 运行时库,可以从 NVIDIA 获取。
手动配置
rust-cublas 的构建脚本默认会尝试通过 pkg-config 定位 cublas
。在某些情况下,这可能不起作用,例如:
- 在没有 pkg-config 的系统上,
- 交叉编译时,或
- cuBLAS 没有安装到默认的系统库目录(例如
/usr/lib
)。
因此,可以通过导出以下环境变量来配置构建脚本
-
CUBLAS_LIB_DIR
如果指定,将使用此目录来查找 cuBLAS 运行时库,例如/opt/cuda/lib
-
CUBLAS_STATIC
如果指定,cuBLAS 库将作为静态链接而不是动态链接。 -
CUBLAS_LIBS
如果指定,将用于查找具有不同名称的 cuBLAS 库。
如果指定了 CUBLAS_LIB_DIR
或 CUBLAS_INCLUDE_DIR
,则构建脚本将跳过 pkg-config 步骤。
如果您还需要运行编译的二进制文件,请确保它们可用
# Linux; for other platforms consult the instructions that come with cuBLAS
cd <cublas_installpath>
export LD_LIBRARY_PATH=`pwd`:$LD_LIBRARY_PATH
贡献
想要贡献吗?太棒了!我们提供了 指导您开始贡献代码或文档的说明。
我们有一个主要是实时协作的文化,在 Github 和 Collenchyma Gitter 频道 上进行。您还可以联系维护者 {@MJ,@hobofan}。
变更日志
您可以在根目录的 CHANGELOG.md 文件中找到发布历史。
变更日志是记录项目所有变更的日志或记录,如网站或软件项目,通常包括错误修复、新功能等记录 - 维基百科
我们使用 Clog,Rust 自动生成 CHANGELOG 文件的工具。
许可证
根据以下任一项许可
- Apache License,版本 2.0,(LICENSE-APACHE 或 https://apache.ac.cn/licenses/LICENSE-2.0)
- MIT 许可证 (LICENSE-MIT 或 http://opensource.org/licenses/MIT)
任选其一。
贡献
除非您明确声明,否则根据 Apache-2.0 许可证定义的,您有意提交的任何贡献,均将根据上述方式双许可,而无需附加条款或条件。
依赖
~160KB