36 lines
567 B
Nix
36 lines
567 B
Nix
|
{ pkgs, ... }:
|
||
|
|
|
||
|
|
pkgs.mkShell {
|
||
|
|
packages = with pkgs; [
|
||
|
|
(python312.withPackages (p: [
|
||
|
|
p.numpy
|
||
|
|
p.scikit-learn
|
||
|
|
p.scipy
|
||
|
|
p.pandas
|
||
|
|
p.matplotlib
|
||
|
|
p.torch-bin
|
||
|
|
# p.torchvision
|
||
|
|
]))
|
||
|
|
libffi
|
||
|
|
openssl
|
||
|
|
stdenv.cc.cc
|
||
|
|
linuxPackages.nvidia_x11
|
||
|
|
binutils
|
||
|
|
cudatoolkit
|
||
|
|
libGLU
|
||
|
|
libGL
|
||
|
|
];
|
||
|
|
LD_LIBRARY_PATH = pkgs.lib.makeLibraryPath (
|
||
|
|
with pkgs;
|
||
|
|
[
|
||
|
|
stdenv.cc.cc
|
||
|
|
linuxPackages.nvidia_x11
|
||
|
|
binutils
|
||
|
|
cudatoolkit
|
||
|
|
libGLU
|
||
|
|
libGL
|
||
|
|
]
|
||
|
|
);
|
||
|
|
CUDA_PATH = pkgs.cudatoolkit;
|
||
|
|
}
|