blob: bab1e7d2a5665f7e1b023a22f6bcd820d1caf783 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
|
{ pkgs ? (import <nixpkgs> {}).pkgs }:
with pkgs;
mkShell {
buildInputs = [
python312Packages.numpy
python312Packages.transformers
python312Packages.torch
];
shellHook = ''
# fixes libstdc++ issues and libgl.so issues
LD_LIBRARY_PATH=${stdenv.cc.cc.lib}/lib/:/run/opengl-driver/lib/
'';
}
|