文件名称:Rust 机器学习库
文件大小:243KB
文件格式:ZIP
更新时间:2024-07-18 12:42:40
rust machine-learning cuda neural-networks
签名Rust 机器学习库正在进行维护特征便携式加速计算在支持 Vulkan/Metal/DX12 的 GPU 上运行 SPIR-V 着色器与 ndarray 互操作,Tensor 模拟 Array轻量级异步/非阻塞 API目前使用 GLSL 作为着色器语言。当 rust-gpu 获得足够的计算着色器支持时,就有可能用 Rust 编写可移植的 GPU 代码!平台Linux / Unix使用 Vulkan 支持 GPU。在 Ubuntu 18.04 AMD RX 580 / NV GTX 1060 上测试macOS / iOS支持带有 Metal 的 GPU。计划支持 Vulkan。GPU 执行未经测试。视窗支持带有 DX12 的 GPU。计划支持 Vulkan。在 Windows 10、AMD RX 580 上测试。注意:使用cargo test -- --num-threads 1运行windows 测试以避免在太多线程上创建太多gpu 实例。跨线程共享访问是安全的,但为多个进程中的每一个创建一个设备可能会失败。数据集鸢尾花https://gist.github.com/curran/