diff options
author | hacker1024 <hacker1024@users.sourceforge.net> | 2023-08-15 00:11:04 +0000 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-08-15 00:11:04 +0000 |
commit | 9fe4aee53a47b6e3971491f47fb6613f7b9c25aa (patch) | |
tree | 5fcf565844c76a018b7ad313de2f302af6b87c79 /pkgs/development/python-modules/torch | |
parent | ecfbcc51f2a9579c9a7660d9bf817c4e929305c4 (diff) |
python3Packages.torch: Allow CUDA 8.7
Diffstat (limited to 'pkgs/development/python-modules/torch')
-rw-r--r-- | pkgs/development/python-modules/torch/default.nix | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/pkgs/development/python-modules/torch/default.nix b/pkgs/development/python-modules/torch/default.nix index 912628bf94973..6e56df8b183d8 100644 --- a/pkgs/development/python-modules/torch/default.nix +++ b/pkgs/development/python-modules/torch/default.nix @@ -68,7 +68,7 @@ let # https://github.com/pytorch/pytorch/blob/v2.0.1/torch/utils/cpp_extension.py#L1744 supportedTorchCudaCapabilities = let - real = ["3.5" "3.7" "5.0" "5.2" "5.3" "6.0" "6.1" "6.2" "7.0" "7.2" "7.5" "8.0" "8.6" "8.9" "9.0"]; + real = ["3.5" "3.7" "5.0" "5.2" "5.3" "6.0" "6.1" "6.2" "7.0" "7.2" "7.5" "8.0" "8.6" "8.7" "8.9" "9.0"]; ptx = lists.map (x: "${x}+PTX") real; in real ++ ptx; |