From 376bace04fdcc24c8a4d886f2b1b9e581cbdc754 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Thu, 8 Feb 2024 02:00:42 +0300 Subject: [PATCH] Lock tensordict to 0.2.1 --- installer.py | 4 ++-- requirements.txt | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/installer.py b/installer.py index 137733ec2..aabcc6020 100644 --- a/installer.py +++ b/installer.py @@ -461,11 +461,11 @@ def check_torch(): install(os.environ.get('MKL_PACKAGE', 'mkl==2024.0.0'), 'mkl') install(os.environ.get('DPCPP_PACKAGE', 'mkl-dpcpp==2024.0.0'), 'mkl-dpcpp') else: - if sys.version_info[1] == 11: + if sys.version_info.minor == 11: pytorch_pip = 'https://github.com/Nuullll/intel-extension-for-pytorch/releases/download/v2.1.10%2Bxpu/torch-2.1.0a0+cxx11.abi-cp311-cp311-win_amd64.whl' torchvision_pip = 'https://github.com/Nuullll/intel-extension-for-pytorch/releases/download/v2.1.10%2Bxpu/torchvision-0.16.0a0+cxx11.abi-cp311-cp311-win_amd64.whl' ipex_pip = 'https://github.com/Nuullll/intel-extension-for-pytorch/releases/download/v2.1.10%2Bxpu/intel_extension_for_pytorch-2.1.10+xpu-cp311-cp311-win_amd64.whl' - elif sys.version_info[1] == 10: + elif sys.version_info.minor == 10: pytorch_pip = 'https://github.com/Nuullll/intel-extension-for-pytorch/releases/download/v2.1.10%2Bxpu/torch-2.1.0a0+cxx11.abi-cp310-cp310-win_amd64.whl' torchvision_pip = 'https://github.com/Nuullll/intel-extension-for-pytorch/releases/download/v2.1.10%2Bxpu/torchvision-0.16.0a0+cxx11.abi-cp310-cp310-win_amd64.whl' ipex_pip = 'https://github.com/Nuullll/intel-extension-for-pytorch/releases/download/v2.1.10%2Bxpu/intel_extension_for_pytorch-2.1.10+xpu-cp310-cp310-win_amd64.whl' diff --git a/requirements.txt b/requirements.txt index a12005c7e..e6c9e0b1a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -34,7 +34,7 @@ rich safetensors scipy tb_nightly -tensordict +tensordict==0.2.1 toml torchdiffeq voluptuous