Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
Torchaudio
Commits
be0a61e6
Unverified
Commit
be0a61e6
authored
Jun 23, 2021
by
moto
Committed by
GitHub
Jun 23, 2021
Browse files
[HOT FIX] Temporarily pin nightly version on Linux GPU unit test (#1606)
parent
f25ca476
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
12 deletions
+6
-12
.circleci/unittest/linux/scripts/install.sh
.circleci/unittest/linux/scripts/install.sh
+6
-12
No files found.
.circleci/unittest/linux/scripts/install.sh
View file @
be0a61e6
...
...
@@ -44,20 +44,14 @@ conda activate "${env_dir}"
if
[
"
${
os
}
"
==
MacOSX
]
||
[
-z
"
${
CUDA_VERSION
:-}
"
]
;
then
device
=
"cpu"
printf
"Installing PyTorch with %s
\n
"
"
$device
}"
(
set
-x
pip
install
--pre
torch
==
1.10.0.dev20210618
-f
"https://download.pytorch.org/whl/nightly/
${
device
}
/torch_nightly.html"
)
else
version
=
"
$(
python
-c
"print('.'.join(
\"
${
CUDA_VERSION
}
\"
.split('.')[:2]))"
)
"
cudatoolkit
=
"cudatoolkit=
${
version
}
"
printf
"Installing PyTorch with %s
\n
"
"
${
cudatoolkit
}
"
(
set
-x
conda
install
${
CONDA_CHANNEL_FLAGS
:-}
-y
-c
"pytorch-
${
UPLOAD_CHANNEL
}
"
"pytorch-
${
UPLOAD_CHANNEL
}
::pytorch"
${
cudatoolkit
}
)
device
=
cu
"
$(
python
-c
"print(''.join(
\"
${
CUDA_VERSION
}
\"
.split('.')[:2]))"
)
"
fi
printf
"Installing PyTorch with %s
\n
"
"
${
device
}
"
(
set
-x
pip
install
--pre
torch
==
1.10.0.dev20210618
-f
"https://download.pytorch.org/whl/nightly/
${
device
}
/torch_nightly.html"
)
# 2. Install torchaudio
printf
"* Installing torchaudio
\n
"
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment