gbtb/nix-stable-diffusion
1
1
Fork 0
mirror of https://github.com/gbtb/nix-stable-diffusion.git synced 2024-09-11 11:16:15 +03:00
Code Issues Packages Projects Releases Wiki Activity

switch torch and torchvision to -bin versions for nvidia

Browse Source
This commit is contained in:
atagen 2022-12-04 00:33:14 +11:00
parent ba3e2e7bc9
commit 1a6ff3b2fb
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
flake.nix
View File

@ -177,9 +177,8 @@
};
overlay_nvidia = nixpkgs: pythonPackages:
{
torch = pythonPackages.torch.override {
cudaSupport = true;
};
torch = pythonPackages.torch-bin;
torchvision = pythonPackages.torchvision-bin;
};
in
{