set cuda capaebilities

This commit is contained in:
Johannes Jöns 2024-01-10 19:58:09 +01:00
parent fea6ae70a4
commit 8f7dba8365

View file

@ -50,7 +50,10 @@
}; };
nixpkgs = { nixpkgs = {
config = { cudaSupport = true; }; config = {
cudaSupport = true;
cudaCapabilities = [ "8.6" ];
};
hostPlatform = { hostPlatform = {
system = "x86_64-linux"; system = "x86_64-linux";
config = "x86_64-unknown-linux-gnu"; config = "x86_64-unknown-linux-gnu";