Skip to content

Instantly share code, notes, and snippets.

@Nimdis

Nimdis/sh

Created May 28, 2023 11:24
Show Gist options
  • Save Nimdis/e2285e3002dd8f7e162e9ed86c2f2595 to your computer and use it in GitHub Desktop.
Save Nimdis/e2285e3002dd8f7e162e9ed86c2f2595 to your computer and use it in GitHub Desktop.
Download ZIP
ROCM AMD GPUs troubles fixing
Raw
sh
# if you cannot install pytorch with
pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.1.1
# just replace it with
pip install torch torchvision --index-url https://download.pytorch.org/whl/rocm5.1.1
# don't forget to uninstall torch before installing ROCM version
pip uninstall torch torchvision
# if you get "Segmentation fault (core dumped)" just use
export HSA_OVERRIDE_GFX_VERSION=10.3.0
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment