38 lines
1.0 KiB
Nix
38 lines
1.0 KiB
Nix
{ callPackage
|
|
, cudatoolkit_10_1, cudatoolkit_10_2
|
|
, cudatoolkit_11_0, cudatoolkit_11_1, cudatoolkit_11_2
|
|
}:
|
|
|
|
rec {
|
|
cutensor_cudatoolkit_10_1 = callPackage ./generic.nix rec {
|
|
version = "1.2.2.5";
|
|
libPath = "lib/10.1";
|
|
cudatoolkit = cudatoolkit_10_1;
|
|
# 1.2.2 is compatible with CUDA 11.0, 11.1, and 11.2:
|
|
# ephemeral doc at https://developer.nvidia.com/cutensor/downloads
|
|
sha256 = "1dl9bd71frhac9cb8lvnh71zfsnqxbxbfhndvva2zf6nh0my4klm";
|
|
};
|
|
|
|
cutensor_cudatoolkit_10_2 = cutensor_cudatoolkit_10_1.override {
|
|
libPath = "lib/10.2";
|
|
cudatoolkit = cudatoolkit_10_2;
|
|
};
|
|
|
|
cutensor_cudatoolkit_10 = cutensor_cudatoolkit_10_2;
|
|
|
|
cutensor_cudatoolkit_11_0 = cutensor_cudatoolkit_10_2.override {
|
|
libPath = "lib/11";
|
|
cudatoolkit = cudatoolkit_11_0;
|
|
};
|
|
|
|
cutensor_cudatoolkit_11_1 = cutensor_cudatoolkit_11_0.override {
|
|
cudatoolkit = cudatoolkit_11_1;
|
|
};
|
|
|
|
cutensor_cudatoolkit_11_2 = cutensor_cudatoolkit_11_0.override {
|
|
cudatoolkit = cudatoolkit_11_2;
|
|
};
|
|
|
|
cutensor_cudatoolkit_11 = cutensor_cudatoolkit_11_2;
|
|
}
|