search

CaptchaAgent / hcaptcha-model-factory

🏗 hCaptcha image label binary model factory (PyTorch Training, Cluster-based Auto Label Tools, Export ONNX model, ONNX model inference)
GNU General Public License v3.0
72 stars 18 forks source link

[question] Turn on CUDA #47

Closed bitcoinautomatic closed 1 year ago

bitcoinautomatic commented 1 year ago

Hello. I can't turn on CUDA. Everything that needed to be installed, it works fine on cpu, but I don’t know how to rearrange it on gpu.

beiyuouo commented 1 year ago

You need to install the GPU version of PyTorch. You can follow: https://pytorch.org/get-started/locally/