Unverified Commit 9cf78b87 authored by Pavol Rusnak's avatar Pavol Rusnak
Browse files

cudaPackages.gpus: disable Compute Capability 10.0a, 10.1, 10.1a and 12.0a to save space 

RTX 50{7,8,9}0 is Compute Capability 12.0
parent 80632f16

pkgs/development/cuda-modules/gpus.nix

+8 −5
Original line number Diff line number Diff line
@@ -26,6 +26,9 @@ 
# Many thanks to Arnon Shimoni for maintaining a list of these architectures and capabilities.

# Without your work, this would have been much more difficult.

# https://arnon.dk/matching-sm-architectures-arch-and-gencode-for-various-nvidia-cards/

#

# https://en.wikipedia.org/wiki/CUDA#GPUs_supported



[

  {

    # Tesla K40
@@ -199,7 +202,7 @@ 
    computeCapability = "10.0a";

    isJetson = false;

    minCudaVersion = "12.8";

    dontDefaultAfter = null;

    dontDefaultAfter = "12.0"; # disable to reduce size of OnnxRuntime and Torch CUDA binaries

    maxCudaVersion = null;

  }

  {
@@ -208,7 +211,7 @@ 
    computeCapability = "10.1";

    isJetson = false;

    minCudaVersion = "12.8";

    dontDefaultAfter = null;

    dontDefaultAfter = "12.0"; # disable to reduce size of OnnxRuntime and Torch CUDA binaries

    maxCudaVersion = null;

  }

  {
@@ -217,7 +220,7 @@ 
    computeCapability = "10.1a";

    isJetson = false;

    minCudaVersion = "12.8";

    dontDefaultAfter = null;

    dontDefaultAfter = "12.0"; # disable to reduce size of OnnxRuntime and Torch CUDA binaries

    maxCudaVersion = null;

  }

  {
@@ -235,7 +238,7 @@ 
    computeCapability = "12.0a";

    isJetson = false;

    minCudaVersion = "12.8";

    dontDefaultAfter = null;

    dontDefaultAfter = "12.0"; # disable to reduce size of OnnxRuntime and Torch CUDA binaries

    maxCudaVersion = null;

  }

]