Files
nixpkgs/pkgs/development/python-modules/pytorch
Daniël de Kok 6038b56de8 python3Packages.pytorch: add compute capabilities for CUDA 11
CUDA 11 supports capabilities 8.0 and 8.6. The change adds these
capabilities when CUDA 11 is used, enabling support for Ampere GPUs.
2021-03-04 11:16:48 +01:00
..