Support ROCm

[githust66]

I have already installed torch==2.4.0+rocm6.3.2, but when I execute pip install sam2, it still tries to download other versions of torch.Below is my environmental information.

PyTorch version: 2.4.0+rocm6.3.2
Is debug build: False
CUDA used to build PyTorch: N/A
ROCM used to build PyTorch: 6.3.42134-a9a80e791

OS: Ubuntu 22.04.5 LTS (x86_64)
GCC version: (conda-forge gcc 12.1.0-17) 12.1.0
Clang version: Could not collect
CMake version: version 3.22.1
Libc version: glibc-2.35

Python version: 3.10.14 (main, May 6 2024, 19:42:50) [GCC 11.2.0] (64-bit runtime)
Python platform: Linux-5.15.167.4-microsoft-standard-WSL2-x86_64-with-glibc2.35
Is CUDA available: True
CUDA runtime version: Could not collect
CUDA_MODULE_LOADING set to: LAZY
GPU models and configuration: AMD Radeon RX 7900 XT (gfx1100)
Nvidia driver version: Could not collect
cuDNN version: Could not collect
HIP runtime version: 6.3.42134
MIOpen runtime version: 3.3.0
Is XNNPACK available: True

CPU:
Architecture: x86_64
CPU op-mode(s): 32-bit, 64-bit
Address sizes: 48 bits physical, 48 bits virtual
Byte Order: Little Endian
CPU(s): 8
On-line CPU(s) list: 0-7
Vendor ID: AuthenticAMD
Model name: AMD Ryzen 7 7700 8-Core Processor
CPU family: 25
Model: 97
Thread(s) per core: 2
Core(s) per socket: 4
Socket(s): 1
Stepping: 2
BogoMIPS: 7599.84

Versions of relevant libraries:
[pip3] numpy==1.26.4
[pip3] onnx==1.17.0
[pip3] onnxruntime-gpu==1.20.1
[pip3] onnxruntime-rocm==1.19.0
[pip3] onnxruntime-training==1.19.0+rocm632.66
[pip3] open_clip_torch==2.30.0
[pip3] optree==0.14.0
[pip3] pytorch-lightning==2.5.0.post0
[pip3] pytorch-triton-rocm==3.0.0+rocm6.3.2.75cc27c26a
[pip3] torch==2.4.0+rocm6.3.2
[pip3] torchaudio==2.4.0+rocm6.3.2
[pip3] torchmetrics==1.6.1
[pip3] torchsde==0.2.6
[pip3] torchvision==0.19.0+rocm6.3.2
[conda] numpy 1.26.4 pypi_0 pypi
[conda] open-clip-torch 2.30.0 pypi_0 pypi
[conda] optree 0.14.0 pypi_0 pypi
[conda] pytorch-lightning 2.5.0.post0 pypi_0 pypi
[conda] pytorch-triton-rocm 3.0.0+rocm6.3.2.75cc27c26a pypi_0 pypi
[conda] torch 2.4.0+rocm6.3.2 pypi_0 pypi
[conda] torchaudio 2.4.0+rocm6.3.2 pypi_0 pypi
[conda] torchmetrics 1.6.1 pypi_0 pypi
[conda] torchsde 0.2.6 pypi_0 pypi
[conda] torchvision 0.19.0+rocm6.3.2 pypi_0 pypi

[Zarxrax]

I dont know if its the cause of your problem, but I believe pytorch 2.5.1 or greater is required.

[githust66]

I dont know if its the cause of your problem, but I believe pytorch 2.5.1 or greater is required.

I tried torch 2.5.1 + rocm6.3.2, The same result applies to downloading and installing other torches without rocm. Executing pip install sam 2 installs torch 2.6.0, Executing pip install sam 2 = = 0.4.1 installs torch 2.3.1, I'll post the log for pip install sam 2 later