| Summary: | sci-libs/pytorch-2.0.0 request for rocm use flag, to build for AMD's rocm runtime | ||
|---|---|---|---|
| Product: | Gentoo Linux | Reporter: | Lorenzo Bertini <lorenzobertini97> |
| Component: | Current packages | Assignee: | Tupone Alfredo <tupone> |
| Status: | UNCONFIRMED --- | ||
| Severity: | enhancement | CC: | alex, bratzmeister, eugene.shalygin, gentoobugs, lorenzobertini97, sur3 |
| Priority: | Normal | ||
| Version: | unspecified | ||
| Hardware: | All | ||
| OS: | All | ||
| Whiteboard: | |||
| Package list: | Runtime testing required: | --- | |
|
Description
Lorenzo Bertini
2023-04-29 20:39:27 UTC
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 |
