nixsearch / pkg / python38Packages.torchgpipe / 0.0.5
python38Packages.torchgpipe 0.0.5
GPipe implemented in Pytorch and optimized for CUDA rather than TPU
Linux
last changed
Pin this version
$ devbox add python38Packages.torchgpipe@0.0.5
$ nix shell github:NixOS/nixpkgs/83162ab3b97d0e13b08e28938133381a7515c1e3#python38Packages.torchgpipe
Per system
| System | Attribute | Commit | Date | Outputs | Flags |
|---|---|---|---|---|---|
x86_64-linuxLinux · x86-64 |
python38Packages.torchgpipe |
83162ab3b97d |
JSON: /v2/pkg?name=python38Packages.torchgpipe · /v2/resolve?name=python38Packages.torchgpipe&version=0.0.5