search

CoinCheung / pytorch-loss

label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful
MIT License
2.17k stars 374 forks source link

need cuda for installing on MacOs #14

Closed reymondzzzz closed 3 years ago

reymondzzzz commented 3 years ago

Can you fix setup.py for installing for cpu version MacOS?

CoinCheung commented 3 years ago

Hi,

Currently these code is only tested on linux (which is my working environment). I am afraid I cannot transfer the cuda code to mac OS in short future, but if you do not want to use cuda based code, you can copy-and-paste the pure pytorch codes to your project. It is easy to find it out by using grep command, please have try :)