cudaPackages.saxpy: now available pre-11.4 with CUDA Toolkit
This commit is contained in:
parent
c27ade81bf
commit
501a1af970
@ -72,10 +72,7 @@ let
|
|||||||
|
|
||||||
# Loose packages
|
# Loose packages
|
||||||
cudatoolkit = final.callPackage ../development/cuda-modules/cudatoolkit {};
|
cudatoolkit = final.callPackage ../development/cuda-modules/cudatoolkit {};
|
||||||
# SaxPy is only available after 11.4 because it requires redistributable versions of CUDA libraries.
|
saxpy = final.callPackage ../development/cuda-modules/saxpy {};
|
||||||
saxpy = attrsets.optionalAttrs (strings.versionAtLeast cudaVersion "11.4") (
|
|
||||||
final.callPackage ../development/cuda-modules/saxpy {}
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
# NCCL is not supported on Jetson, because it does not use NVLink or PCI-e for inter-GPU communication.
|
# NCCL is not supported on Jetson, because it does not use NVLink or PCI-e for inter-GPU communication.
|
||||||
# https://forums.developer.nvidia.com/t/can-jetson-orin-support-nccl/232845/9
|
# https://forums.developer.nvidia.com/t/can-jetson-orin-support-nccl/232845/9
|
||||||
|
Loading…
Reference in New Issue
Block a user