We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3772fed commit 6d5b1faCopy full SHA for 6d5b1fa
1 file changed
.github/workflows/scripts/pytorch-install.sh
@@ -20,8 +20,8 @@ pip install typing-extensions==4.12.2
20
echo $MATRIX_CUDA_VERSION
21
echo $MATRIX_TORCH_VERSION
22
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']]; \
+minv = {'2.2': 118, '2.3': 118, '2.4': 118, '2.5': 118, '2.6': 118, '2.7': 118}[env['MATRIX_TORCH_VERSION']]; \
+maxv = {'2.2': 121, '2.3': 121, '2.4': 124, '2.5': 124, '2.6': 126, '2.7': 128}[env['MATRIX_TORCH_VERSION']]; \
25
print(max(min(int(env['MATRIX_CUDA_VERSION']), maxv), minv))" \
26
)
27
if [[ ${pytorch_version} == *"dev"* ]]; then
0 commit comments