File tree 1 file changed +5
-0
lines changed
1 file changed +5
-0
lines changed Original file line number Diff line number Diff line change @@ -13,6 +13,11 @@ if [[ "$working_dir" == "$FBGEMM_REPO" ]]; then cd fbgemm_gpu || echo "Failed to
13
13
# # Build clean/wheel will be done in pre-script. Set flag such that setup.py will skip these steps in Nova workflow
14
14
export BUILD_FROM_NOVA=1
15
15
16
+ # # Override existing ENV VAR in Nova for Release
17
+ export CHANNEL=" test"
18
+ export PIP_INSTALL_TORCH=" pip install torch --index-url https://download.pytorch.org/whl/${CU_VERSION} "
19
+ export PYTORCH_S3_BUCKET_PATH=" s3://pytorch/whl/${CU_VERSION} /"
20
+
16
21
# # Overwrite existing ENV VAR in Nova
17
22
if [[ " $CONDA_ENV " != " " ]]; then export CONDA_RUN=" conda run --no-capture-output -p ${CONDA_ENV} " && echo " $CONDA_RUN " ; fi
18
23
if [[ " $CU_VERSION " == " cu118" ]]; then export TORCH_CUDA_ARCH_LIST=' 7.0;8.0' && echo " $TORCH_CUDA_ARCH_LIST " ; fi
You can’t perform that action at this time.
0 commit comments