File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -12,11 +12,12 @@ export CONDA_ENV="build_binary"
12
12
if [[ ${MATRIX_PYTHON_VERSION} = ' 3.13t' ]]; then
13
13
# use conda-forge to install python3.13t
14
14
conda create -y -n " ${CONDA_ENV} " python=" 3.13" python-freethreading -c conda-forge
15
+ conda run -n " ${CONDA_ENV} " python -c " import sys; print(f'python GIL enabled: {sys._is_gil_enabled()}')"
15
16
else
16
17
conda create -y -n " ${CONDA_ENV} " python=" ${MATRIX_PYTHON_VERSION} "
17
18
fi
18
19
19
- conda run -n build_binary python --version
20
+ conda run -n " ${CONDA_ENV} " python --version
20
21
21
22
# Install pytorch, torchrec and fbgemm as per
22
23
# installation instructions on following page
@@ -83,7 +84,7 @@ conda run -n "${CONDA_ENV}" pip install fbgemm-gpu --index-url "$PYTORCH_URL"
83
84
conda run -n " ${CONDA_ENV} " pip install torchmetrics==1.0.3
84
85
85
86
# install tensordict from pypi
86
- conda run -n " ${CONDA_ENV} " pip install tensordict==0.7 .1
87
+ conda run -n " ${CONDA_ENV} " pip install tensordict==0.8 .1
87
88
88
89
# install torchrec
89
90
conda run -n " ${CONDA_ENV} " pip install torchrec --index-url " $PYTORCH_URL "
You can’t perform that action at this time.
0 commit comments