diff --git a/packages.yaml b/packages.yaml index 96017412328b57843eb222ac1d8db60a51b2086e..1dd5e099c40089e3fe392b8c10520693e83dc82b 100644 --- a/packages.yaml +++ b/packages.yaml @@ -4,3 +4,6 @@ packages: compiler: [gcc@10.3.0] python: version: [3.8.11] + py-torch: + version: [1.11.0] + variants: [~cuda~rocm~valgrind~mkldnn~mpi~gloo+tensorpipe~onnx_ml] diff --git a/spack.yaml b/spack.yaml index 6b8b2943abfbc5f0c95827f103fb33a381f5fa39..37d8d2ca0bcb85843dfaf9c1d9313ccb67407c50 100644 --- a/spack.yaml +++ b/spack.yaml @@ -55,7 +55,6 @@ spack: - py-hippounit@1.3.6 - py-efel@4.0.4 - py-hbp-neuromorphic-platform@0.10.1 - - py-torch~cuda~rocm~valgrind~mkldnn~mpi~gloo+tensorpipe~onnx_ml@1.11.0:^protobuf@:3.17.999 - py-lfpy@2.3 - py-elephant - py-frites