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

CUDA Version？ #19

Open LouisSu123 opened 2 years ago

LouisSu123 commented 2 years ago

I don`t know the cuda version to run this project? Maybe conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=10.2 -c pytorch?