File tree 1 file changed +2
-2
lines changed
1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 93
93
$cudaVersionPytorch = $cudaVersion.Substring(0, $cudaVersion.Length - 1)
94
94
if ([int]$cudaVersionPytorch -gt 118) { $pytorchVersion = "torch==2.2.0" } else {$pytorchVersion = "torch==2.0.1"}
95
95
python -m pip install --upgrade --no-cache-dir $pytorchVersion+cu$cudaVersionPytorch --index-url https://download.pytorch.org/whl/cu$cudaVersionPytorch
96
- python -m pip install build setuptools wheel ninja
96
+ python -m pip install build setuptools wheel ninja requests
97
97
98
98
# Print version information
99
99
python --version
@@ -206,7 +206,7 @@ jobs:
206
206
sudo apt-get update
207
207
sudo apt-get install -y --no-install-recommends rocsparse-dev rocthrust-dev rocblas-dev hipblas-dev hipsparse-dev
208
208
209
- python -m pip install --upgrade build setuptools wheel
209
+ python -m pip install --upgrade build setuptools wheel requests
210
210
211
211
if [[ "${{ matrix.rocm }}" == "5.7.1" ]]; then
212
212
python -m pip install torch==2.2.0 --index-url https://download.pytorch.org/whl/rocm5.7
You can’t perform that action at this time.
0 commit comments