/pytorch

Fork of pytorch that builds on Python 2.7, CUDA 11.2, CuDNN 8.1.1, compute capability up to 8.6 (Ampere GPUs)

Primary LanguageC++OtherNOASSERTION

Stargazers

No one’s star this repository yet.