6个版本 (2个稳定版)
2.0.3 | 2024年6月19日 |
---|---|
2.0.1 | 2024年6月5日 |
0.1.3 | 2024年6月5日 |
0.1.2 | 2024年5月19日 |
#483 in 机器学习
每月下载量443次
44KB
677 行
RayBNN_Neural
使用CUDA、OpenCL和oneAPI通过GPU、CPU和FPGA在Rust中实现稀疏权重的神经网络
安装Arrayfire
在https://arrayfire.com/binaries/安装Arrayfire 3.9.0的二进制文件
或从源代码构建https://github.com/arrayfire/arrayfire/wiki/Getting-ArrayFire
添加到Cargo.toml
arrayfire = { version = "3.8.1", package = "arrayfire_fork" }
rayon = "1.10.0"
num = "0.4.3"
num-traits = "0.2.19"
half = { version = "2.4.1" , features = ["num-traits"] }
RayBNN_DataLoader = "2.0.3"
RayBNN_Graph = "2.0.3"
RayBNN_Sparse = "2.0.2"
RayBNN_Optimizer = "2.0.1"
RayBNN_Neural = "2.0.3"
示例列表
依赖项
~8.5MB
~162K SLoC