diff --git a/tests/conda-envs/cpu.yaml b/tests/conda-envs/cpu.yaml index b769970..3d70190 100644 --- a/tests/conda-envs/cpu.yaml +++ b/tests/conda-envs/cpu.yaml @@ -2,6 +2,7 @@ name: cpu channels: - pytorch dependencies: + - python==3.11 - pytorch - torchvision - torchaudio diff --git a/tests/conda-envs/cuda.yaml b/tests/conda-envs/cuda.yaml index 06c0afa..f83cfb4 100644 --- a/tests/conda-envs/cuda.yaml +++ b/tests/conda-envs/cuda.yaml @@ -3,6 +3,7 @@ channels: - pytorch - nvidia dependencies: + - python==3.11 - pytorch - torchvision - torchaudio diff --git a/tests/conda-envs/ipex.yaml b/tests/conda-envs/ipex.yaml index d136704..d6d5b2d 100644 --- a/tests/conda-envs/ipex.yaml +++ b/tests/conda-envs/ipex.yaml @@ -3,6 +3,7 @@ channels: - intel - conda-forge dependencies: + - python==3.11 - intel-aikit-pytorch - pytorch>=2.0.1=*_xpu_* - intel-extension-for-pytorch