45 ms·
Still waiting for better rocm support without docker. I wish it was at least at the level of tensorflow.
by oehtXRwMkIs 7y ago
Still waiting for better rocm support without docker. I wish it was at least at the level of tensorflow.
- t-vi 6y agoYou can compile PyTorch this pretty easily, I wrote a recipe here: https://lernapparat.de/pytorch-rocm/ https://lernapparat.de/pytorch-rocm/ No docker involved, just Debian + the ROCm packages from their repository. I'm doing this relatively regularly, and it has worked well for me for the past half year or so. I guess I could upload wheels somewhere, but I'm never sure how tight the dependency on ROCm has to be. Over the last release cycle, PyTorch/ROCm gained support for TorchVision including GPU ops among other things (based on support for extension compiled via setup.py). Master has support for loading extensions that are compiled just in time. If you find things not working as well as they should, don't hesitate to tag me on a bug report or send a mail. (The PyTorch/ROCm is mostly for my own entertainment, so I don't speak for anyone.)
- sytelus 7y agoI'd love better cross-platform distributed support. It's ok to be non-performant or not support fency backends on Windows but at least your interfaces should be available cross-platform.