hiyouga/pytorch

•Updated over 1 year ago

Base image with pytorch and flash attention

Image

Machine learning & AI

10K+

hiyouga/pytorch repository overview

No overview available

This repository doesn't have an overview

Recent tags

Content type

Image

Digest

sha256:bb606e5cb…

Size

8.3 GB

Last updated

over 1 year ago

docker pull hiyouga/pytorch:th2.6.0-cu124-flashattn2.7.4-cxx11abi0-devel