pyrox.dev
Merge pull request #234085 from ConnorBaker/fix/pytorch_2_0_1-supported-cuda-targets
python3Packages.torch: update CUDA capabilities for v2.0.1 release
authored by Samuel Ainsworth and committed by GitHub 80c63c04 ef247999
+2
-2
pkgs/development/python-modules/torch/default.nix
+2
-2
pkgs/development/python-modules/torch/default.nix
···
65
65
let
66
66
setBool = v: if v then "1" else "0";
67
67
68
-
# https://github.com/pytorch/pytorch/blob/v1.13.1/torch/utils/cpp_extension.py#L1751
68
+
# https://github.com/pytorch/pytorch/blob/v2.0.1/torch/utils/cpp_extension.py#L1744
69
69
supportedTorchCudaCapabilities =
70
70
let
71
-
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"];
71
+
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"];
72
72
ptx = lists.map (x: "${x}+PTX") real;
73
73
in
74
74
real ++ ptx;