File name
Commit message
Commit date
2024-11-02
{
inputs = {
nixpkgs.url = "github:NixOS/nixpkgs/nixos-unstable";
flake-utils.url = "github:numtide/flake-utils";
flake-utils.inputs.nixpkgs.follows = "nixpkgs";
};
outputs =
{ nixpkgs, flake-utils, ... }:
flake-utils.lib.eachDefaultSystem (
system:
let
pkgs = import nixpkgs {
inherit system;
config.allowUnfree = true;
# `cudnn_8_9` is marked as broken but still seems to work as before
config.allowBroken = true;
};
in
{
devShells = {
default = pkgs.mkShell {
nativeBuildInputs = with pkgs; [
act
ffmpeg-full
go-task
parallel
pv
python312
rsync
websocat
uv
cudaPackages_12.cudnn
cudaPackages_12.libcublas
];
# https://github.com/NixOS/nixpkgs/issues/278976#issuecomment-1879685177
# NOTE: Without adding `/run/...` the following error occurs
# RuntimeError: CUDA failed with error CUDA driver version is insufficient for CUDA runtime version
#
# NOTE: sometimes it still doesn't work but rebooting the system fixes it
LD_LIBRARY_PATH = "/run/opengl-driver/lib:${
pkgs.lib.makeLibraryPath [
pkgs.cudaPackages_12.cudnn
pkgs.cudaPackages_12.libcublas
pkgs.zlib
pkgs.stdenv.cc.cc
pkgs.openssl
]
}";
shellHook = ''
source .venv/bin/activate
source .env
'';
};
};
formatter = pkgs.nixfmt;
}
);
}