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
8b2fbf28
Unverified
Commit
8b2fbf28
authored
Oct 12, 2022
by
Andrey Talman
Committed by
GitHub
Oct 12, 2022
Browse files
Fix torchaudio build channel for the release (#2759)
* Fix torchaudio build channel * Fix channel
parent
720c36b1
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
4 deletions
+4
-4
packaging/pkg_helpers.bash
packaging/pkg_helpers.bash
+4
-4
No files found.
packaging/pkg_helpers.bash
View file @
8b2fbf28
...
...
@@ -189,7 +189,7 @@ setup_pip_pytorch_version() {
if
[[
-z
"
$PYTORCH_VERSION
"
]]
;
then
# Install latest prerelease version of torch, per our nightlies, consistent
# with the requested cuda version
pip_install
--pre
torch
-f
"https://download.pytorch.org/whl/
nightly
/
${
WHEEL_DIR
}
torch_
nightly
.html"
pip_install
--pre
torch
-f
"https://download.pytorch.org/whl/
test
/
${
WHEEL_DIR
}
torch_
test
.html"
# CUDA and CPU are ABI compatible on the CPU-only parts, so strip in this case
export
PYTORCH_VERSION
=
"
$(
pip show torch |
grep
^Version: |
sed
's/Version: *//'
|
sed
's/+.\+//'
)
"
else
...
...
@@ -206,11 +206,11 @@ setup_pip_pytorch_version() {
setup_conda_pytorch_constraint
()
{
CONDA_CHANNEL_FLAGS
=
"
${
CONDA_CHANNEL_FLAGS
}
"
if
[[
-z
"
$PYTORCH_VERSION
"
]]
;
then
export
CONDA_CHANNEL_FLAGS
=
"
${
CONDA_CHANNEL_FLAGS
}
-c pytorch-
nightly
"
export
CONDA_CHANNEL_FLAGS
=
"
${
CONDA_CHANNEL_FLAGS
}
-c pytorch-
test
"
if
[[
"
$OSTYPE
"
==
"msys"
]]
;
then
export
PYTORCH_VERSION
=
"
$(
conda search
--json
-c
pytorch-
nightly
pytorch | python
-c
"import sys, json; data=json.load(sys.stdin); print(data['pytorch'][-1]['version'])"
)
"
export
PYTORCH_VERSION
=
"
$(
conda search
--json
-c
pytorch-
test
pytorch | python
-c
"import sys, json; data=json.load(sys.stdin); print(data['pytorch'][-1]['version'])"
)
"
else
export
PYTORCH_VERSION
=
"
$(
conda search
--json
'pytorch[channel=pytorch-
nightly
]'
| python3
-c
"import sys, json, re; print(re.sub(r'
\\
+.*
$'
, '', json.load(sys.stdin)['pytorch'][-1]['version']))"
)
"
export
PYTORCH_VERSION
=
"
$(
conda search
--json
'pytorch[channel=pytorch-
test
]'
| python3
-c
"import sys, json, re; print(re.sub(r'
\\
+.*
$'
, '', json.load(sys.stdin)['pytorch'][-1]['version']))"
)
"
fi
else
export
CONDA_CHANNEL_FLAGS
=
"
${
CONDA_CHANNEL_FLAGS
}
-c pytorch -c pytorch-test -c pytorch-nightly"
...
...
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