3 ms·
Or if you are using Nix use https://github.com/nixvital/ml-pkgs https://github.com/nixvital/ml-pkgs outputs = { self, nixpkgs, ... }@inputs: { overlays =
by breakds 3y ago
Or if you are using Nix use https://github.com/nixvital/ml-pkgs https://github.com/nixvital/ml-pkgs
outputs = { self, nixpkgs, ... }@inputs: {
overlays = {
dev = nixpkgs.lib.composeManyExtensions [
inputs.ml-pkgs.overlays.torch-family
# Add some other overlays
];
};
};
in your flake.nix and you can use pytorch 2.0.1 compiled with CUDA 11.8 running on your 4090s. Downside is that the first time you will have to compile it and it can be quite long.