mirror of
https://github.com/nix-community/dream2nix.git
synced 2024-08-17 10:20:33 +03:00
override/python: fix devShells for nvidia packages
This commit is contained in:
parent
306e5bff07
commit
6ddb9a7f5e
@ -2,7 +2,7 @@
|
||||
description = "My flake with dream2nix packages";
|
||||
|
||||
inputs = {
|
||||
dream2nix.url = "github:nix-community/dream2nix/pythonEnv";
|
||||
dream2nix.url = "github:nix-community/dream2nix";
|
||||
nixpkgs.follows = "dream2nix/nixpkgs";
|
||||
flake-parts.url = "github:hercules-ci/flake-parts";
|
||||
};
|
||||
|
5
overrides/python/nvidia-cublas-cu12/default.nix
Normal file
5
overrides/python/nvidia-cublas-cu12/default.nix
Normal file
@ -0,0 +1,5 @@
|
||||
{
|
||||
imports = [
|
||||
./remove-nvidia-init.nix
|
||||
];
|
||||
}
|
10
overrides/python/nvidia-cublas-cu12/remove-nvidia-init.nix
Normal file
10
overrides/python/nvidia-cublas-cu12/remove-nvidia-init.nix
Normal file
@ -0,0 +1,10 @@
|
||||
{
|
||||
config,
|
||||
lib,
|
||||
...
|
||||
}: {
|
||||
mkDerivation.postFixup =
|
||||
# prevents conflicts in nixpkgs buildEnv for python
|
||||
lib.mkIf (lib.hasSuffix ".whl" config.mkDerivation.src)
|
||||
"rm $out/lib/*/site-packages/nvidia/__pycache__/__init__.*";
|
||||
}
|
5
overrides/python/nvidia-cuda-cupti-cu12/default.nix
Normal file
5
overrides/python/nvidia-cuda-cupti-cu12/default.nix
Normal file
@ -0,0 +1,5 @@
|
||||
{
|
||||
imports = [
|
||||
../nvidia-cublas-cu12/remove-nvidia-init.nix
|
||||
];
|
||||
}
|
5
overrides/python/nvidia-cuda-nvrtc-cu12/default.nix
Normal file
5
overrides/python/nvidia-cuda-nvrtc-cu12/default.nix
Normal file
@ -0,0 +1,5 @@
|
||||
{
|
||||
imports = [
|
||||
../nvidia-cublas-cu12/remove-nvidia-init.nix
|
||||
];
|
||||
}
|
5
overrides/python/nvidia-cuda-runtime-cu12/default.nix
Normal file
5
overrides/python/nvidia-cuda-runtime-cu12/default.nix
Normal file
@ -0,0 +1,5 @@
|
||||
{
|
||||
imports = [
|
||||
../nvidia-cublas-cu12/remove-nvidia-init.nix
|
||||
];
|
||||
}
|
5
overrides/python/nvidia-cudnn-cu12/default.nix
Normal file
5
overrides/python/nvidia-cudnn-cu12/default.nix
Normal file
@ -0,0 +1,5 @@
|
||||
{
|
||||
imports = [
|
||||
../nvidia-cublas-cu12/remove-nvidia-init.nix
|
||||
];
|
||||
}
|
5
overrides/python/nvidia-cufft-cu12/default.nix
Normal file
5
overrides/python/nvidia-cufft-cu12/default.nix
Normal file
@ -0,0 +1,5 @@
|
||||
{
|
||||
imports = [
|
||||
../nvidia-cublas-cu12/remove-nvidia-init.nix
|
||||
];
|
||||
}
|
5
overrides/python/nvidia-curand-cu12/default.nix
Normal file
5
overrides/python/nvidia-curand-cu12/default.nix
Normal file
@ -0,0 +1,5 @@
|
||||
{
|
||||
imports = [
|
||||
../nvidia-cublas-cu12/remove-nvidia-init.nix
|
||||
];
|
||||
}
|
5
overrides/python/nvidia-cusolver-cu12/default.nix
Normal file
5
overrides/python/nvidia-cusolver-cu12/default.nix
Normal file
@ -0,0 +1,5 @@
|
||||
{
|
||||
imports = [
|
||||
../nvidia-cublas-cu12/remove-nvidia-init.nix
|
||||
];
|
||||
}
|
5
overrides/python/nvidia-cusparse-cu12/default.nix
Normal file
5
overrides/python/nvidia-cusparse-cu12/default.nix
Normal file
@ -0,0 +1,5 @@
|
||||
{
|
||||
imports = [
|
||||
../nvidia-cublas-cu12/remove-nvidia-init.nix
|
||||
];
|
||||
}
|
5
overrides/python/nvidia-nccl-cu12/default.nix
Normal file
5
overrides/python/nvidia-nccl-cu12/default.nix
Normal file
@ -0,0 +1,5 @@
|
||||
{
|
||||
imports = [
|
||||
../nvidia-cublas-cu12/remove-nvidia-init.nix
|
||||
];
|
||||
}
|
5
overrides/python/nvidia-nvjitlink-cu12/default.nix
Normal file
5
overrides/python/nvidia-nvjitlink-cu12/default.nix
Normal file
@ -0,0 +1,5 @@
|
||||
{
|
||||
imports = [
|
||||
../nvidia-cublas-cu12/remove-nvidia-init.nix
|
||||
];
|
||||
}
|
Loading…
Reference in New Issue
Block a user