torch python3-torch | python3-torch-cuda | python3-torch-rocm
