How to install Pytorch with cuda support through CondaPkg

Specific Domains Machine Learning
1

Hi All,

I would like to ask, if anyone knows how to install pytorch with cuda support through CondaPkg. It has to happen that I need to call some language models and I have opted to use transformers from HuggingFace, as sad as it can be. But I have spent quite few days figuring out, how to install pytorch with cuda support, because if I do just ] conda add pytorch, it CondaPkg will install torch without CUDA support. I have ended manually editing .CondaPkg/pixi.toml, add line

[system-requirements]
cuda = "12.4"

and than invoking

~/.julia/artifacts/cefba4912c2b400756d043a2563ef77a0088866b/bin/pixi add pytorch transformers accelerate

in the .CondaPkg directory. Ugly hack indeed it is, but it made it work at least. But if possible, I would like to know the correct solution.

Thanks for answer.
Best wishes.