docker.io/javirub/flashattention-pytorch

该Docker镜像包含PyTorch环境以及FlashAttention库,可用于在PyTorch框架下进行高效的注意力机制相关的深度学习模型训练与推理任务,简化了FlashAttention的部署与使用流程。

下载:1 版本:1
检测到您正在使用广告拦截插件,本站为公益站点,依赖广告维持运转 🙏 查看如何关闭 ×