Commit 3a5b47e2 authored by DGdev91's avatar DGdev91

Forcing PyTorch version for AMD GPUs automatic install

The old code tries to install the newest versions of pytorch, wich is currently 2.0. Forcing it to 1.13.1
parent 22bcc7be
......@@ -118,7 +118,7 @@ case "$gpu_info" in
esac
if echo "$gpu_info" | grep -q "AMD" && [[ -z "${TORCH_COMMAND}" ]]
then
export TORCH_COMMAND="pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.2"
export TORCH_COMMAND="pip install torch==1.13.1+rocm5.2 torchvision==0.14.1+rocm5.2 --extra-index-url https://download.pytorch.org/whl/rocm5.2"
fi
for preq in "${GIT}" "${python_cmd}"
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment