@@ -146,9 +146,9 @@ function install_torchaudio() {
146146 if [[ " $1 " == " cuda" ]]; then
147147 # TODO: This is better to be passed as a parameter from _linux-test workflow
148148 # so that it can be consistent with what is set in build
149- TORCH_CUDA_ARCH_LIST=" 8.0;8.6" pip_install --no-use-pep517 --user " git+https://github.com/pytorch/audio.git@${commit} "
149+ TORCH_CUDA_ARCH_LIST=" 8.0;8.6" pip_install --no-use-pep517 " git+https://github.com/pytorch/audio.git@${commit} "
150150 else
151- pip_install --no-use-pep517 --user " git+https://github.com/pytorch/audio.git@${commit} "
151+ pip_install --no-use-pep517 " git+https://github.com/pytorch/audio.git@${commit} "
152152 fi
153153
154154}
@@ -158,8 +158,8 @@ function install_torchtext() {
158158 local text_commit
159159 data_commit=$( get_pinned_commit data)
160160 text_commit=$( get_pinned_commit text)
161- pip_install --no-use-pep517 --user " git+https://github.com/pytorch/data.git@${data_commit} "
162- pip_install --no-use-pep517 --user " git+https://github.com/pytorch/text.git@${text_commit} "
161+ pip_install --no-use-pep517 " git+https://github.com/pytorch/data.git@${data_commit} "
162+ pip_install --no-use-pep517 " git+https://github.com/pytorch/text.git@${text_commit} "
163163}
164164
165165function install_torchvision() {
@@ -172,16 +172,12 @@ function install_torchvision() {
172172 echo ' char* dlerror(void) { return "";}' | gcc -fpic -shared -o " ${HOME} /dlerror.so" -x c -
173173 LD_PRELOAD=${orig_preload} :${HOME} /dlerror.so
174174 fi
175- pip_install --no-use-pep517 --user " git+https://github.com/pytorch/vision.git@${commit} "
175+ pip_install --no-use-pep517 " git+https://github.com/pytorch/vision.git@${commit} "
176176 if [ -n " ${LD_PRELOAD} " ]; then
177177 LD_PRELOAD=${orig_preload}
178178 fi
179179}
180180
181- function install_tlparse() {
182- pip_install --user " tlparse==0.3.25"
183- PATH=" $( python -m site --user-base) /bin:$PATH "
184- }
185181
186182function install_torchrec_and_fbgemm() {
187183 local torchrec_commit
@@ -192,8 +188,8 @@ function install_torchrec_and_fbgemm() {
192188 pip_uninstall fbgemm-gpu-nightly
193189 pip_install setuptools-git-versioning scikit-build pyre-extensions
194190 # See https://github.com/pytorch/pytorch/issues/106971
195- CUDA_PATH=/usr/local/cuda-12.1 pip_install --no-use-pep517 --user " git+https://github.com/pytorch/FBGEMM.git@${fbgemm_commit} #egg=fbgemm-gpu&subdirectory=fbgemm_gpu"
196- pip_install --no-use-pep517 --user " git+https://github.com/pytorch/torchrec.git@${torchrec_commit} "
191+ CUDA_PATH=/usr/local/cuda-12.1 pip_install --no-use-pep517 " git+https://github.com/pytorch/FBGEMM.git@${fbgemm_commit} #egg=fbgemm-gpu&subdirectory=fbgemm_gpu"
192+ pip_install --no-use-pep517 " git+https://github.com/pytorch/torchrec.git@${torchrec_commit} "
197193}
198194
199195function clone_pytorch_xla() {
0 commit comments