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