Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Conversation

@infinity0
Copy link
Contributor

@infinity0 infinity0 commented Nov 27, 2024

With my relatively new AMD Radeon RX 7800 XT I need to set the following envvars for ROCm to work:

export PYTORCH_HIP_ALLOC_CONF=expandable_segments:True
export HSA_OVERRIDE_GFX_VERSION=11.0.0
export HCC_AMDGPU_TARGET=gfx1101

Instead of hardcoding this all, just pass the whole parent environment in, this is the default behaviour anyway for most "spawn" APIs.

Also torchaudio can't detect my ffmpeg 7, it seems to not support this yet. libsox-dev works as an alternative.

Now I can demux stems in 20 seconds compared to 10 minutes for Wine/CPU! :D

@infinity0 infinity0 mentioned this pull request Nov 27, 2024
@iffyloop
Copy link
Member

iffyloop commented Dec 2, 2024

Interesting... I suppose that was naive of me to not expect people to need to customize environment variables for their GPU. I feel like there was a reason I wanted to isolate the Demucs process from the rest of the system environment variables (which are kind of messy on Windows) but I'm not sure that my concern was actually valid. I'm hesitant to push out a new release and notify everyone to download 1-2GB update for this change only, but I'll merge this PR in on the next release.

Thank you so much for submitting this, and I'm happy to hear it's working now on your ROCm Linux device!

@iffyloop iffyloop changed the base branch from main to dev January 27, 2025 04:28
@iffyloop iffyloop merged commit 08d23f7 into stemrollerapp:dev Jan 27, 2025
@iffyloop
Copy link
Member

Merged! Thank you again for helping to improve StemRoller @infinity0!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants