search

dqshuai / MetaFormer

A PyTorch implementation of "MetaFormer: A Unified Meta Framework for Fine-Grained Recognition". A reference PyTorch implementation of “CoAtNet: Marrying Convolution and Attention for All Data Sizes”
MIT License
210 stars 36 forks source link

Apex and Cuda Version #26

Closed RorryB closed 1 year ago

RorryB commented 1 year ago

Ran into issues installing Apex. Just wanted to post a note that the install worked with torch 1.5.1, CUDA 10.1, and rolling apex back via the comment here https://github.com/NVIDIA/apex/issues/1043#issuecomment-773868833