Skip to content
代码片段 群组 项目
提交 4da92281 编辑于 作者: AUTOMATIC1111's avatar AUTOMATIC1111
浏览文件

pin version for torch for Navi3 according to comment from #11228

上级 ed855783
No related branches found
No related tags found
无相关合并请求
......@@ -134,7 +134,7 @@ case "$gpu_info" in
*"Navi 2"*) export HSA_OVERRIDE_GFX_VERSION=10.3.0
;;
*"Navi 3"*) [[ -z "${TORCH_COMMAND}" ]] && \
export TORCH_COMMAND="pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm5.5"
export TORCH_COMMAND="pip install --pre torch==2.1.0.dev-20230614+rocm5.5 torchvision==0.16.0.dev-20230614+rocm5.5 --index-url https://download.pytorch.org/whl/nightly/rocm5.5"
# Navi 3 needs at least 5.5 which is only on the nightly chain
;;
*"Renoir"*) export HSA_OVERRIDE_GFX_VERSION=9.0.0
......
0% 加载中 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册