We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 117f11a commit cfdaaaaCopy full SHA for cfdaaaa
.ci/pytorch/common_utils.sh
@@ -152,6 +152,12 @@ function get_pinned_commit() {
152
function install_torchaudio() {
153
local commit
154
commit=$(get_pinned_commit audio)
155
+ # TODO (huydhn): PyTorch CI docker image set the default TORCH_CUDA_ARCH_LIST
156
+ # to Maxwell. This default doesn't make sense anymore and should be cleaned up
157
+ if [[ "${BUILD_ENVIRONMENT}" == *cuda* ]]; then
158
+ TORCH_CUDA_ARCH_LIST=$(nvidia-smi --query-gpu=compute_cap --format=csv | tail -n 1)
159
+ export TORCH_CUDA_ARCH_LIST
160
+ fi
161
pip_build_and_install "git+https://github.com/pytorch/audio.git@${commit}" dist/audio
162
}
163
0 commit comments