feat: neuro: some
This commit is contained in:
@@ -113,12 +113,22 @@
|
|||||||
zramSwap.swapDevices = 1;
|
zramSwap.swapDevices = 1;
|
||||||
zramSwap.memoryPercent = lib.mkDefault 100;
|
zramSwap.memoryPercent = lib.mkDefault 100;
|
||||||
|
|
||||||
environment.systemPackages = with pkgs; [
|
environment.systemPackages = with pkgs; let
|
||||||
# (pkgs.python3.withPackages (ps: with ps; [
|
python-ai = python3.withPackages (ps: let
|
||||||
# hectic.py3-openai-shap-e
|
torchCuda = ps.torchWithCuda;
|
||||||
# torchWithCuda torchvision pytorch3d
|
torchvision = ps.torchvision.override { torch = torchCuda; };
|
||||||
# fvcore iopath tqdm
|
pytorch3dCuda = ps.pytorch3d.override { torch = torchCuda; };
|
||||||
# ]))
|
in [
|
||||||
|
torchCuda
|
||||||
|
torchvision
|
||||||
|
pytorch3dCuda
|
||||||
|
ps.fvcore
|
||||||
|
ps.iopath
|
||||||
|
ps.tqdm
|
||||||
|
hectic.py3-openai-shap-e # Uncomment when needed; depends on torch
|
||||||
|
]);
|
||||||
|
in [
|
||||||
|
python-ai
|
||||||
git
|
git
|
||||||
neovim
|
neovim
|
||||||
wget
|
wget
|
||||||
|
|||||||
Reference in New Issue
Block a user