Hlavní navigace

CuPy 9 s podporou GPU AMD

Sdílet

Jan Fikar 30. 4. 2021
CuPy Autor: CuPy Team

CuPy je open source matematická knihovna pro Python, která používá array kompatibilní s NumPy a je akcelerována na GPU NVidia pomocí CUDA (odtud název). Minulý týden vyšla knihovna CuPy ve verzi 9, která přidává podporu pro GPU AMD pomocí ROCm 4.0. Použití s AMD je stále experimentální, zájemci se mohou podívat do návodu.

Mezi další novinky patří JIT API ke psaní kernelů CUDA přímo v Pythonu. Nově je vyžadována CUDA alespoň 9.2 a Python alespoň 3.6. Zdrojové kódy CuPy jsou na GitHubu.

(zdroj: phoronix)

Našli jste v článku chybu?
  • Aktualita je stará, nové názory již nelze přidávat.