Skip to content
This repository was archived by the owner on Sep 10, 2025. It is now read-only.

Add check for rocminfo in the installation script to replace the nightly URL with the appropriate ROCm whl. #1279

Merged
merged 2 commits into from
Oct 7, 2024
Merged

Add check for rocminfo in the installation script to replace the nightly URL with the appropriate ROCm whl. #1279

Changes from 1 commit
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
cc2d6c6
Add check for rocminfo in the installation script to replace the nigh…
Oct 7, 2024
0e06e0f
Update install/install_requirements.sh
vmpuri Oct 7, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions install/install_requirements.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -67,6 +67,9 @@ TUNE_NIGHTLY_VERSION=dev20240928
if [[ -x "$(command -v nvidia-smi)" ]];
then
TORCH_NIGHTLY_URL="https://download.pytorch.org/whl/nightly/cu121"
elif [[ -x "$(command -v rocminfo)" ]];
then
TORCH_NIGHTLY_URL="https://download.pytorch.org/whl/rocm6.1"
else
TORCH_NIGHTLY_URL="https://download.pytorch.org/whl/nightly/cpu"
fi
Expand Down
Loading