Merge 57722f86e6a6e76818887472b18b6c9636c512da into 2174ce5afea90ca489d222f539988dcef59f1027

This commit is contained in:
Christian Bayle 2025-05-03 02:17:19 -04:00 committed by GitHub
commit c24e8db761
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -152,6 +152,10 @@ case "$gpu_info" in
;; ;;
*"Navi 2"*) export HSA_OVERRIDE_GFX_VERSION=10.3.0 *"Navi 2"*) export HSA_OVERRIDE_GFX_VERSION=10.3.0
;; ;;
*"Navi 3"*"RX 76"*) [[ -z "${TORCH_COMMAND}" ]] && \
export TORCH_COMMAND="pip install torch torchvision --index-url https://download.pytorch.org/whl/rocm5.7" && \
export HSA_OVERRIDE_GFX_VERSION=11.0.0
;;
*"Navi 3"*) [[ -z "${TORCH_COMMAND}" ]] && \ *"Navi 3"*) [[ -z "${TORCH_COMMAND}" ]] && \
export TORCH_COMMAND="pip install torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm5.7" export TORCH_COMMAND="pip install torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm5.7"
;; ;;