"vscode:/vscode.git/clone" did not exist on "5788c85518ed44f3cd4b0de057b3263f4e59b835"
Unverified Commit e8ef4c08 authored by Woosuk Kwon's avatar Woosuk Kwon Committed by GitHub
Browse files

Fix PyTorch index URL in workflow (#1378)

parent 348897af
......@@ -6,7 +6,7 @@ cuda_version=$3
# Install torch
$python_executable -m pip install numpy pyyaml scipy ipython mkl mkl-include ninja cython typing pandas typing-extensions dataclasses setuptools && conda clean -ya
$python_executable -m pip install torch==${pytorch_version}+cu${cuda_version//./} --index-url https://download.pytorch.org/whl/cu${cuda_version//./}
$python_executable -m pip install torch==${pytorch_version}+cu${cuda_version//./} --extra-index-url https://download.pytorch.org/whl/cu${cuda_version//./}
# Print version information
$python_executable --version
......
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