File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 6363 if ($IsLinux) {
6464 mamba install -y --channel-priority flexible --override-channels -c $cudaChannel "${cudaChannel}::cuda-toolkit=$cudaVersion" "${cudaChannel}::cuda-nvcc_linux-64" "${cudaChannel}::cuda-cccl" "${cudaChannel}::cuda-cudart" "${cudaChannel}::cuda-cudart-dev"
6565 } elseif ($IsWindows) {
66- mamba install -y --channel-priority flexible --override-channels -c $cudaChannel "${cudaChannel}::cuda-toolkit=$cudaVersion " "${cudaChannel}::cuda-nvcc_win-64 " "${cudaChannel}::cuda-cccl " "${cudaChannel}::cuda-cudart" "${cudaChannel}::cuda-cudart-dev"
66+ mamba install -y --channel-priority flexible --override-channels -c $cudaChannel "${cudaChannel}::cuda-nvcc_win-64 " "${cudaChannel}::cuda-cccl " "${cudaChannel}::cuda-libraries-dev=$cudaVersion " "${cudaChannel}::cuda-cudart" "${cudaChannel}::cuda-cudart-dev"
6767 } else {
6868 throw 'Unsupported CUDA wheel build platform'
6969 }
You can’t perform that action at this time.
0 commit comments