@@ -127,9 +127,9 @@ function install_torchaudio() {
127127 if [[ " $1 " == " cuda" ]]; then
128128 # TODO: This is better to be passed as a parameter from _linux-test workflow
129129 # so that it can be consistent with what is set in build
130- TORCH_CUDA_ARCH_LIST=" 8.0;8.6" pip_install --no-use-pep517 --user " git+https://github.com/pytorch/audio.git@${commit} "
130+ TORCH_CUDA_ARCH_LIST=" 8.0;8.6" pip_install --no-use-pep517 " git+https://github.com/pytorch/audio.git@${commit} "
131131 else
132- pip_install --no-use-pep517 --user " git+https://github.com/pytorch/audio.git@${commit} "
132+ pip_install --no-use-pep517 " git+https://github.com/pytorch/audio.git@${commit} "
133133 fi
134134
135135}
@@ -139,8 +139,8 @@ function install_torchtext() {
139139 local text_commit
140140 data_commit=$( get_pinned_commit data)
141141 text_commit=$( get_pinned_commit text)
142- pip_install --no-use-pep517 --user " git+https://github.com/pytorch/data.git@${data_commit} "
143- pip_install --no-use-pep517 --user " git+https://github.com/pytorch/text.git@${text_commit} "
142+ pip_install --no-use-pep517 " git+https://github.com/pytorch/data.git@${data_commit} "
143+ pip_install --no-use-pep517 " git+https://github.com/pytorch/text.git@${text_commit} "
144144}
145145
146146function install_torchvision() {
@@ -153,16 +153,12 @@ function install_torchvision() {
153153 echo ' char* dlerror(void) { return "";}' | gcc -fpic -shared -o " ${HOME} /dlerror.so" -x c -
154154 LD_PRELOAD=${orig_preload} :${HOME} /dlerror.so
155155 fi
156- pip_install --no-use-pep517 --user " git+https://github.com/pytorch/vision.git@${commit} "
156+ pip_install --no-use-pep517 " git+https://github.com/pytorch/vision.git@${commit} "
157157 if [ -n " ${LD_PRELOAD} " ]; then
158158 LD_PRELOAD=${orig_preload}
159159 fi
160160}
161161
162- function install_tlparse() {
163- pip_install --user " tlparse==0.3.30"
164- PATH=" $( python -m site --user-base) /bin:$PATH "
165- }
166162
167163function install_torchrec_and_fbgemm() {
168164 local torchrec_commit
@@ -178,7 +174,7 @@ function install_torchrec_and_fbgemm() {
178174
179175 if [[ " $BUILD_ENVIRONMENT " == * rocm* ]] ; then
180176 # install torchrec first because it installs fbgemm nightly on top of rocm fbgemm
181- pip_install --no-use-pep517 --user " git+https://github.com/pytorch/torchrec.git@${torchrec_commit} "
177+ pip_install --no-use-pep517 " git+https://github.com/pytorch/torchrec.git@${torchrec_commit} "
182178 pip_uninstall fbgemm-gpu-nightly
183179
184180 pip_install tabulate # needed for newer fbgemm
@@ -195,8 +191,8 @@ function install_torchrec_and_fbgemm() {
195191 rm -rf fbgemm
196192 else
197193 # See https://github.com/pytorch/pytorch/issues/106971
198- 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"
199- pip_install --no-use-pep517 --user " git+https://github.com/pytorch/torchrec.git@${torchrec_commit} "
194+ 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"
195+ pip_install --no-use-pep517 " git+https://github.com/pytorch/torchrec.git@${torchrec_commit} "
200196 fi
201197}
202198
@@ -239,7 +235,7 @@ function checkout_install_torchbench() {
239235function install_torchao() {
240236 local commit
241237 commit=$( get_pinned_commit torchao)
242- pip_install --no-use-pep517 --user " git+https://github.com/pytorch/ao.git@${commit} "
238+ pip_install --no-use-pep517 " git+https://github.com/pytorch/ao.git@${commit} "
243239}
244240
245241function print_sccache_stats() {
0 commit comments