search

ryujaehun / pytorch-gpu-benchmark

Using the famous cnn model in Pytorch, we run benchmarks on various gpu.
MIT License
226 stars 85 forks source link

bugfix , support for torch 2.0 #22

Closed hellojixian closed 1 year ago

hellojixian commented 1 year ago

this code will work for pyTorch>=2.0

devrandom commented 1 year ago

tested ACK