torch python3-torch | python3-torch-cuda
