Skip to content

Commit 6d5b1fa

Browse files
authored
Fixed(MInference): fix the release torch version (#161)
1 parent 3772fed commit 6d5b1fa

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.github/workflows/scripts/pytorch-install.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,8 @@ pip install typing-extensions==4.12.2
2020
echo $MATRIX_CUDA_VERSION
2121
echo $MATRIX_TORCH_VERSION
2222
export TORCH_CUDA_VERSION=$(python -c "from os import environ as env; \
23-
minv = {'2.4': 118, '2.5': 118, '2.6': 118, '2.7': 118}[env['MATRIX_TORCH_VERSION']]; \
24-
maxv = {'2.4': 124, '2.5': 124, '2.6': 126, '2.7': 128}[env['MATRIX_TORCH_VERSION']]; \
23+
minv = {'2.2': 118, '2.3': 118, '2.4': 118, '2.5': 118, '2.6': 118, '2.7': 118}[env['MATRIX_TORCH_VERSION']]; \
24+
maxv = {'2.2': 121, '2.3': 121, '2.4': 124, '2.5': 124, '2.6': 126, '2.7': 128}[env['MATRIX_TORCH_VERSION']]; \
2525
print(max(min(int(env['MATRIX_CUDA_VERSION']), maxv), minv))" \
2626
)
2727
if [[ ${pytorch_version} == *"dev"* ]]; then

0 commit comments

Comments
 (0)