AMD MI GPUs supported

cd ~/github/

docker run -it --rm \
  --device=/dev/kfd --device=/dev/dri \
  --group-add video \
  --shm-size=32g \
  --name dust3r \
  -v $(pwd):/workspace -w /workspace \
  rocm/pytorch:rocm6.4.3_ubuntu24.04_py3.12_pytorch_release_2.6.0

cd dust3r 
conda create -n dust3r python=3.12 cmake=3.14.0
source /opt/conda/etc/profile.d/conda.sh && conda activate dust3r

# # official requries conda install pytorch torchvision pytorch-cuda=12.1 -c pytorch -c nvidia 
pip3 install --pre torch==2.8.0 torchvision==0.23.0 torchaudio==2.8.0 --index-url https://download.pytorch.org/whl/rocm6.4  

# bypass existing modules update
EXCLUDE_PKGS="torch|torchvision|torchaudio"
grep -vE -i '^('"$EXCLUDE_PKGS"')([<>=!~;[:space:]]|$)' requirements.txt > requirements.tmp && mv requirements.tmp requirements.txt
python3 -m pip install -r requirements.txt 
python3 -m pip install -r requirements_optional.txt

# croco kernel build 
export PYTORCH_ROCM_ARCH="gfx942"
cd croco/models/curope/
# check https://github.com/naver/croco/pull/48 to update setpy.py 
python setup.py build_ext --inplace
cd ../../../


## download weights
mkdir -p checkpoints/
wget https://download.europe.naverlabs.com/ComputerVision/DUSt3R/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth -P checkpoints/

# run sample
python3 quick_start.py