Skip to content

Commit 16f6124

Browse files
committed
update tensordict version
1 parent df23042 commit 16f6124

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

.github/scripts/validate_binaries.sh

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,12 @@ export CONDA_ENV="build_binary"
1212
if [[ ${MATRIX_PYTHON_VERSION} = '3.13t' ]]; then
1313
# use conda-forge to install python3.13t
1414
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()}')"
1516
else
1617
conda create -y -n "${CONDA_ENV}" python="${MATRIX_PYTHON_VERSION}"
1718
fi
1819

19-
conda run -n build_binary python --version
20+
conda run -n "${CONDA_ENV}" python --version
2021

2122
# Install pytorch, torchrec and fbgemm as per
2223
# installation instructions on following page
@@ -83,7 +84,7 @@ conda run -n "${CONDA_ENV}" pip install fbgemm-gpu --index-url "$PYTORCH_URL"
8384
conda run -n "${CONDA_ENV}" pip install torchmetrics==1.0.3
8485

8586
# 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
8788

8889
# install torchrec
8990
conda run -n "${CONDA_ENV}" pip install torchrec --index-url "$PYTORCH_URL"

0 commit comments

Comments
 (0)