История изменений
Исправление praseodim, (текущая версия) :
И еще, для некоторых программ еще может понадобиться
apt install rocm-hip-sdk
Если используешь jupyter-notebook-и всякие, их тоже надо ставить из под venv с нужным pytorch или как-то трахаться с настройкой выбора нужного ядра. Мне проще из под venv показалось. И еще, точно не помню, но кажется лучше jupyter ставить ДО установки pytorch
source pytorcht/hip/bin/activate
pip install jupyter notebook
torchvision torchaudio много для чего тоже понадобится. Сейчас nightly можно опустить. Но вообще советую попробовать 6.0, хотя сам не пробовал.
Плюс дополнительные замечания от создателей pytorch по работе с AMD HIP https://pytorch.org/docs/stable/notes/hip.html
Исправление praseodim, :
И еще, для некоторых программ еще может понадобиться
apt install rocm-hip-sdk
Если используешь jupyter-notebook-и всякие, их тоже надо ставить из под venv с нужным pytorch или как-то трахаться с настройкой выбора нужного ядра. Мне проще из под venv показалось.
source pytorcht/hip/bin/activate
pip install jupyter notebook
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm5.7
torchvision torchaudio много для чего тоже понадобится. Сейчас nightly можно опустить. Но вообще советую попробовать 6.0, хотя сам не пробовал.
Плюс дополнительные замечания от создателей pytorch по работе с AMD HIP https://pytorch.org/docs/stable/notes/hip.html
Исправление praseodim, :
И еще, для некоторых программ еще может понадобиться
apt install rocm-hip-sdk
Если используешь jupyter-notebook-и всякие, их тоже надо ставить из под venv с нужным pytorch или как-то трахаться с настройкой выбора нужного ядра. Мне проще из под venv показалось.
source pytorcht/hip/bin/activate
pip install jupyter notebook
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm5.7
torchvision torchaudio много для чего тоже понадобится. Сейчас nightly можно опустить. Но вообще советую попробовать 6.0, хотя сам не пробовал.
Плюс, некоторые замечания от создателей pytorch по работе с AMD HIP https://pytorch.org/docs/stable/notes/hip.html
Исходная версия praseodim, :
И еще, для некоторых программ еще может понадобиться
apt install rocm-hip-sdk
Если используешь jupyter-notebook-и всякие, их тоже надо ставить из под venv с нужным pytorch или как-то трахаться с настройкой выбора нужного ядра. Мне проще из под venv показалось.
source pytorcht/hip/bin/activate
pip install jupyter notebook
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm5.7
torchvision torchaudio много для чего тоже понадобится. Сейчас nightly можно опустить. Но вообще советую попробовать 6.0, хотя сам не пробовал.