Basically this https://bugs.gentoo.org/705712 (I'm not the requester) but for pytorch. Thanks in advance
patches (or pull request are very welcome)
(In reply to Tupone Alfredo from comment #1) > patches (or pull request are very welcome) Sorry for the long delay. A quick look here https://github.com/pytorch/pytorch#from-source reveals that one just needs to run tools/amd_build/build_amd.py to hipify the build. However, the script operates on caffe2, which we package separately. Could you kindly point me in the right direction? Many thanks and best wishes
please do this for current version (too). pytorch >=2.0 should work with ROCm 5.7.1 https://wiki.gentoo.org/wiki/ROCm and has official upstream support, like Lorenzo mentioned. Currently I'm handicapped as the official binaries don´t support my GPU and gentoo doesnt incorporate pytorch with rocm although it has both in portage
Maybe unrelated, but having libcaffe2 as separate package means I can update pytorch and not caffe2 to have mismatched versions. My pytorch real version is then incorrect. What was the reationale behind this separation?
there is a lot of work going on for ROCm 5.7.1 in https://github.com/gentoo/gentoo/pull/33400. Possibly it will be ready to be used with pytorch after this PR
I've seen rocm use flag for pytorch being mentioned on the rocm wiki page but as of now I don't see an ebuild utilising it https://wiki.gentoo.org/wiki/ROCm#ROCm_libraries
(In reply to bratzmeister from comment #6) > I've seen rocm use flag for pytorch being mentioned on the rocm wiki page > but as of now I don't see an ebuild utilising it > https://wiki.gentoo.org/wiki/ROCm#ROCm_libraries on sci-libs/caffe2 (part of pytorch) there is a rocm use flag that do (should) what you asked
