-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathshell.nix
More file actions
32 lines (27 loc) · 853 Bytes
/
shell.nix
File metadata and controls
32 lines (27 loc) · 853 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
{ pkgs ? import <nixpkgs> { } }:
with pkgs; mkShell {
name = "ml-friendly-shell";
buildInputs = [
glib
libGL
glibc
zlib
python39
];
shellHook = ''
# Setup python virtual environment
python -m venv ~/.venv
source ~/.venv/bin/activate
# Standard environment
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:${pkgs.stdenv.cc.cc.lib}/lib"
# [CV2] Software / Hardware bridge
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:${libGL}/lib"
# [CV2] Useful data types
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:${pkgs.glib.out}/lib"
# [Torch (CUDA)] OpenGL
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/run/opengl-driver/lib"
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/run/opengl-driver-32/lib"
# [Numpy] Data compression
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:${pkgs.zlib.outPath}/lib"
'';
}