diff --git a/conda/environments/all_cuda-120_arch-x86_64.yaml b/conda/environments/all_cuda-120_arch-x86_64.yaml index 5d0bb33c4e..13d581d29c 100644 --- a/conda/environments/all_cuda-120_arch-x86_64.yaml +++ b/conda/environments/all_cuda-120_arch-x86_64.yaml @@ -10,7 +10,7 @@ dependencies: - c-compiler - cmake>=3.26.4 - cuda-cudart-dev -- cuda-nvcc=12.0 +- cuda-nvcc - cuda-profiler-api - cuda-python>=12.0,<13.0a0 - cuda-version=12.0 @@ -35,6 +35,7 @@ dependencies: - ipython - joblib>=0.11 - libcublas-dev +- libcufft-dev - libcumlprims==23.8.* - libcurand-dev - libcusolver-dev diff --git a/conda/environments/cpp_all_cuda-120_arch-x86_64.yaml b/conda/environments/cpp_all_cuda-120_arch-x86_64.yaml index 8417cd41fc..4ab81103e5 100644 --- a/conda/environments/cpp_all_cuda-120_arch-x86_64.yaml +++ b/conda/environments/cpp_all_cuda-120_arch-x86_64.yaml @@ -10,7 +10,7 @@ dependencies: - c-compiler - cmake>=3.26.4 - cuda-cudart-dev -- cuda-nvcc=12.0 +- cuda-nvcc - cuda-profiler-api - cuda-version=12.0 - cxx-compiler @@ -18,6 +18,7 @@ dependencies: - gmock>=1.13.0 - gtest>=1.13.0 - libcublas-dev +- libcufft-dev - libcumlprims==23.8.* - libcurand-dev - libcusolver-dev