diff options
author | Todd Gamblin <tgamblin@llnl.gov> | 2022-07-30 15:19:18 -0700 |
---|---|---|
committer | Todd Gamblin <tgamblin@llnl.gov> | 2022-07-31 13:29:20 -0700 |
commit | f52f6e99dbf1131886a80112b8c79dfc414afb7c (patch) | |
tree | 05cb7d64b2395922f2f24683da49f472075be12c /var/spack/repos/builtin/packages/cuda/package.py | |
parent | 549ba1ed32372c67fc57271cde3797d58b7dec6e (diff) | |
download | spack-f52f6e99dbf1131886a80112b8c79dfc414afb7c.tar.gz spack-f52f6e99dbf1131886a80112b8c79dfc414afb7c.tar.bz2 spack-f52f6e99dbf1131886a80112b8c79dfc414afb7c.tar.xz spack-f52f6e99dbf1131886a80112b8c79dfc414afb7c.zip |
black: reformat entire repository with black
Diffstat (limited to 'var/spack/repos/builtin/packages/cuda/package.py')
-rw-r--r-- | var/spack/repos/builtin/packages/cuda/package.py | 580 |
1 files changed, 423 insertions, 157 deletions
diff --git a/var/spack/repos/builtin/packages/cuda/package.py b/var/spack/repos/builtin/packages/cuda/package.py index 01be188ae4..fd76aabebd 100644 --- a/var/spack/repos/builtin/packages/cuda/package.py +++ b/var/spack/repos/builtin/packages/cuda/package.py @@ -24,114 +24,374 @@ from spack.package import * # format returned by platform.system() and 'arch' by platform.machine() _versions = { - '11.7.0': { - 'Linux-aarch64': ('e777839a618ca9a3d5ad42ded43a1b6392af2321a7327635a4afcc986876a21b', 'https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux_sbsa.run'), - 'Linux-x86_64': ('087fdfcbba1f79543b1f78e43a8dfdac5f6db242d042dde820e16dc185892f26', 'https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run'), - 'Linux-ppc64le': ('74a507ac54067c258e6b7c9063c98d411116ecc5c5397b1f6e6a999e86dff08a', 'https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux_ppc64le.run')}, - '11.6.2': { - 'Linux-aarch64': ('b20c014c6bba36b13c50da167ad42e9bd1cea24f3b6297b495ea129c0889f36e', 'https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux_sbsa.run'), - 'Linux-x86_64': ('99b7a73dcc52a52cef4c1fceb4a60c3015ac9b6404082c1677d9efdaba1d4593', 'https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux.run'), - 'Linux-ppc64le': ('869232ff8dbf295a71609738ac9e1b0079ca75597b427f1c026f42b36896afe8', 'https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux_ppc64le.run')}, - '11.6.1': { - 'Linux-aarch64': ('80586b003d58030004d465f5331dc69ee26c95a29516fb2488ff10f034139cb2', 'https://developer.download.nvidia.com/compute/cuda/11.6.1/local_installers/cuda_11.6.1_510.47.03_linux_sbsa.run'), - 'Linux-x86_64': ('ab219afce00b74200113269866fbff75ead037bcfc23551a8338c2684c984d7e', 'https://developer.download.nvidia.com/compute/cuda/11.6.1/local_installers/cuda_11.6.1_510.47.03_linux.run'), - 'Linux-ppc64le': ('ef762efbc00b67d572823c6ec338cc2c0cf0c096f41e6bce18e8d4501f260956', 'https://developer.download.nvidia.com/compute/cuda/11.6.1/local_installers/cuda_11.6.1_510.47.03_linux_ppc64le.run')}, - '11.6.0': { - 'Linux-aarch64': ('5898579f5e59b708520883cb161089f5e4f3426158d1e9f973c49d224085d1d2', 'https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda_11.6.0_510.39.01_linux_sbsa.run'), - 'Linux-x86_64': ('1783da6d63970786040980b57fa3cb6420142159fc7d0e66f8f05c4905d98c83', 'https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda_11.6.0_510.39.01_linux.run'), - 'Linux-ppc64le': ('c86b866a42baf59ddc6f1f4a79e6d77213c90749e77e574f0e0d796a749ab7d0', 'https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda_11.6.0_510.39.01_linux_ppc64le.run')}, - '11.5.2': { - 'Linux-aarch64': ('31337c8bdc224fa1bd07bc4b6a745798392428118cc8ea0fa4446ee4ad47dd30', 'https://developer.download.nvidia.com/compute/cuda/11.5.2/local_installers/cuda_11.5.2_495.29.05_linux_sbsa.run'), - 'Linux-x86_64': ('74959abf02bcba526f0a3aae322c7641b25da040ccd6236d07038f81997b73a6', 'https://developer.download.nvidia.com/compute/cuda/11.5.2/local_installers/cuda_11.5.2_495.29.05_linux.run'), - 'Linux-ppc64le': ('45c468f430436b3e95d5e485a6ba0ec1fa2b23dc6c551c1307b79996ecf0a7ed', 'https://developer.download.nvidia.com/compute/cuda/11.5.2/local_installers/cuda_11.5.2_495.29.05_linux_ppc64le.run')}, - '11.5.1': { - 'Linux-aarch64': ('73e1d0e97c7fa686efe7e00fb1e5f179372c4eec8e14d4f44ab58d5f6cf57f63', 'https://developer.download.nvidia.com/compute/cuda/11.5.1/local_installers/cuda_11.5.1_495.29.05_linux_sbsa.run'), - 'Linux-x86_64': ('60bea2fc0fac95574015f865355afbf599422ec2c85554f5f052b292711a4bca', 'https://developer.download.nvidia.com/compute/cuda/11.5.1/local_installers/cuda_11.5.1_495.29.05_linux.run'), - 'Linux-ppc64le': ('9e0e494d945634fe8ad3e12d7b91806aa4220ed27487bb211030d651b27c67a9', 'https://developer.download.nvidia.com/compute/cuda/11.5.1/local_installers/cuda_11.5.1_495.29.05_linux_ppc64le.run')}, - '11.5.0': { - 'Linux-aarch64': ('6ea9d520cc956cc751a5ac54f4acc39109627f4e614dd0b1a82cc86f2aa7d8c4', 'https://developer.download.nvidia.com/compute/cuda/11.5.0/local_installers/cuda_11.5.0_495.29.05_linux_sbsa.run'), - 'Linux-x86_64': ('ae0a1693d9497cf3d81e6948943e3794636900db71c98d58eefdacaf7f1a1e4c', 'https://developer.download.nvidia.com/compute/cuda/11.5.0/local_installers/cuda_11.5.0_495.29.05_linux.run'), - 'Linux-ppc64le': ('95baefdc5adf165189407b119861ffb2e9800fd94d7fc81d10fb81ed36dc12db', 'https://developer.download.nvidia.com/compute/cuda/11.5.0/local_installers/cuda_11.5.0_495.29.05_linux_ppc64le.run')}, - '11.4.4': { - 'Linux-aarch64': ('c5c08531e48e8fdc2704fa1c1f7195f2c7edd2ee10a466d0e24d05b77d109435', 'https://developer.download.nvidia.com/compute/cuda/11.4.4/local_installers/cuda_11.4.4_470.82.01_linux_sbsa.run'), - 'Linux-x86_64': ('44545a7abb4b66dfc201dcad787b5e8352e5b7ddf3e3cc5b2e9177af419c25c8', 'https://developer.download.nvidia.com/compute/cuda/11.4.4/local_installers/cuda_11.4.4_470.82.01_linux.run'), - 'Linux-ppc64le': ('c71cd4e6c05fde11c0485369a73e7f356080e7a18f0e3ad7244e8fc03a9dd3e2', 'https://developer.download.nvidia.com/compute/cuda/11.4.4/local_installers/cuda_11.4.4_470.82.01_linux_ppc64le.run')}, - '11.4.3': { - 'Linux-aarch64': ('e02db34a487ea3de3eec9db80efd09f12eb69d55aca686cecaeae96a9747b1d4', 'https://developer.download.nvidia.com/compute/cuda/11.4.3/local_installers/cuda_11.4.3_470.82.01_linux_sbsa.run'), - 'Linux-x86_64': ('749183821ffc051e123f12ebdeb171b263d55b86f0dd7c8f23611db1802d6c37', 'https://developer.download.nvidia.com/compute/cuda/11.4.3/local_installers/cuda_11.4.3_470.82.01_linux.run'), - 'Linux-ppc64le': ('08f29cc3ed0b3b82dd9b007186237be2352bb552f99230c450a25e768f5754ee', 'https://developer.download.nvidia.com/compute/cuda/11.4.3/local_installers/cuda_11.4.3_470.82.01_linux_ppc64le.run')}, - '11.4.2': { - 'Linux-aarch64': ('f2c4a52e06329606c8dfb7c5ea3f4cb4c0b28f9d3fdffeeb734fcc98daf580d8', 'https://developer.download.nvidia.com/compute/cuda/11.4.2/local_installers/cuda_11.4.2_470.57.02_linux_sbsa.run'), - 'Linux-x86_64': ('bbd87ca0e913f837454a796367473513cddef555082e4d86ed9a38659cc81f0a', 'https://developer.download.nvidia.com/compute/cuda/11.4.2/local_installers/cuda_11.4.2_470.57.02_linux.run'), - 'Linux-ppc64le': ('a917c2e53dc13fdda7def71fd40920bf3809d5a2caa3e9acfe377fb9fb22f12d', 'https://developer.download.nvidia.com/compute/cuda/11.4.2/local_installers/cuda_11.4.2_470.57.02_linux_ppc64le.run')}, - '11.4.1': { - 'Linux-aarch64': ('8efa725a41dfd3c0c0f453c2dd535d149154102bf2b791718859417b4f84f922', 'https://developer.download.nvidia.com/compute/cuda/11.4.1/local_installers/cuda_11.4.1_470.57.02_linux_sbsa.run'), - 'Linux-x86_64': ('dd6c339a719989d2518f5d54eeac1ed707d0673f8664ba0c4d4b2af7c3ba0005', 'https://developer.download.nvidia.com/compute/cuda/11.4.1/local_installers/cuda_11.4.1_470.57.02_linux.run'), - 'Linux-ppc64le': ('dd92ca04f76ad938da3480e2901c0e52dbff6028ada63c09071ed9e3055dc361', 'https://developer.download.nvidia.com/compute/cuda/11.4.1/local_installers/cuda_11.4.1_470.57.02_linux_ppc64le.run')}, - '11.4.0': { - 'Linux-aarch64': ('f0c8e80d98a601ddca031b6764459984366008c7d3847e7c7f99b36bd4438e3c', 'https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux_sbsa.run'), - 'Linux-x86_64': ('d219db30f7415a115a4ea22bdbb5984b0a18f7f891cad6074c5da45d223aaa4b', 'https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux.run'), - 'Linux-ppc64le': ('6eb2fd0d9d5bc39fb243b5e1789ff827f325d098cd1fbb828a0499552b9544cc', 'https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux_ppc64le.run')}, - '11.3.1': { - 'Linux-aarch64': ('39990d3da88b21289ac20850bc183f0b66275f32e1f562b551c05843bf506e4c', 'https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux_sbsa.run'), - 'Linux-x86_64': ('ad93ea98efced35855c58d3a0fc326377c60917cb3e8c017d3e6d88819bf2934', 'https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux.run'), - 'Linux-ppc64le': ('220f2c10a21500d62b03c6848c1659ebb3a8e10dc0915ab87b86b397058407c5', 'https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux_ppc64le.run')}, - '11.3.0': { - 'Linux-aarch64': ('f7b284cf055fcf18be9a9aa216e3c7968d3e011446eb0c7200a3353c223ce718', 'https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda_11.3.0_465.19.01_linux_sbsa.run'), - 'Linux-x86_64': ('262da7f77db177b153a6b70b5812210f3f72f00eb608ab9cf2a4582328c4bf5c', 'https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda_11.3.0_465.19.01_linux.run'), - 'Linux-ppc64le': ('c0010107933b575a87e27b1293e5dc32b74201486f4ae2f4c8695ea727d22857', 'https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda_11.3.0_465.19.01_linux_ppc64le.run')}, - '11.2.2': { - 'Linux-aarch64': ('2f915ad631331eebdafaabd971723a60290ae8bb090d771075b9e6a0b28cbae6', 'https://developer.download.nvidia.com/compute/cuda/11.2.2/local_installers/cuda_11.2.2_460.32.03_linux_sbsa.run'), - 'Linux-x86_64': ('0a2e477224af7f6003b49edfd2bfee07667a8148fe3627cfd2765f6ad72fa19d', 'https://developer.download.nvidia.com/compute/cuda/11.2.2/local_installers/cuda_11.2.2_460.32.03_linux.run'), - 'Linux-ppc64le': ('2304ec235fe5d1f8bf75f00dc2c2d11473759dc23428dbbd5fb5040bc8c757e3', 'https://developer.download.nvidia.com/compute/cuda/11.2.2/local_installers/cuda_11.2.2_460.32.03_linux_ppc64le.run')}, - '11.2.1': { - 'Linux-aarch64': ('4b322fa6477d1a2cd2f2f526fa520c0f90bef2c264ef8435cb016bebb5456c5e', 'https://developer.download.nvidia.com/compute/cuda/11.2.1/local_installers/cuda_11.2.1_460.32.03_linux_sbsa.run'), - 'Linux-x86_64': ('1da98cb897cc5f58a7445a4a66ca4f6926867706cb3af58a669cdcd8dc3d17c8', 'https://developer.download.nvidia.com/compute/cuda/11.2.1/local_installers/cuda_11.2.1_460.32.03_linux.run'), - 'Linux-ppc64le': ('b3e8b6cd76872deb3acd050d32e197bc1c655e142b169070f0f9753680461a3f', 'https://developer.download.nvidia.com/compute/cuda/11.2.1/local_installers/cuda_11.2.1_460.32.03_linux_ppc64le.run')}, - '11.2.0': { - 'Linux-aarch64': ('c11dc274660e9b47b0f25ca66861a7406246a7191f1b04d0710515fcac0fa6cd', 'https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda_11.2.0_460.27.04_linux_sbsa.run'), - 'Linux-x86_64': ('9c50283241ac325d3085289ed9b9c170531369de41165ce271352d4a898cbdce', 'https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda_11.2.0_460.27.04_linux.run'), - 'Linux-ppc64le': ('adc3267df5dbfdaf51cb4c9b227ba6bfd979a39d9b4136bba0eba6b1dd2a2731', 'https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda_11.2.0_460.27.04_linux_ppc64le.run')}, - '11.1.1': { - 'Linux-aarch64': ('9ab1dbafba205c06bea8c88e38cdadb3038af19cb56e7b3ba734d3d7a84b8f02', 'https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux_sbsa.run'), - 'Linux-x86_64': ('3eae6727086024925ebbcef3e9a45ad379d8490768fd00f9c2d8b6fd9cd8dd8f', 'https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux.run'), - 'Linux-ppc64le': ('023e571fe26ee829c98138dfc305a92279854aac7d184d255fd58c06c6af3c17', 'https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux_ppc64le.run')}, - '11.1.0': { - 'Linux-aarch64': ('878cbd36c5897468ef28f02da50b2f546af0434a8a89d1c724a4d2013d6aa993', 'https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda_11.1.0_455.23.05_linux_sbsa.run'), - 'Linux-x86_64': ('858cbab091fde94556a249b9580fadff55a46eafbcb4d4a741d2dcd358ab94a5', 'https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda_11.1.0_455.23.05_linux.run'), - 'Linux-ppc64le': ('a561e6f7f659bc4100e4713523b0b8aad6b36aa77fac847f6423e7780c750064', 'https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda_11.1.0_455.23.05_linux_ppc64le.run')}, - '11.0.3': { - 'Linux-aarch64': ('1e24f61f79c1043aa3d1d126ff6158daa03a62a51b5195a2ed5fbe75c3b718f3', 'https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux_sbsa.run'), - 'Linux-x86_64': ('b079c4e408adf88c3f1ffb8418a97dc4227c37935676b4bf4ca0beec6c328cc0', 'https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux.run'), - 'Linux-ppc64le': ('4775b21df004b1433bafff9b48a324075c008509f4c0fe28cd060d042d2e0794', 'https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux_ppc64le.run')}, - '11.0.2': { - 'Linux-aarch64': ('23851e30f7c47a1baad92891abde0adbc783de5962c7480b9725198ceacda4a0', 'https://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux_sbsa.run'), - 'Linux-x86_64': ('48247ada0e3f106051029ae8f70fbd0c238040f58b0880e55026374a959a69c1', 'https://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run'), - 'Linux-ppc64le': ('db06d0f3fbf6f7aa1f106fc921ad1c86162210a26e8cb65b171c5240a3bf75da', 'https://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux_ppc64le.run')}, - '10.2.89': { - 'Linux-x86_64': ('560d07fdcf4a46717f2242948cd4f92c5f9b6fc7eae10dd996614da913d5ca11', 'https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run'), - 'Linux-ppc64le': ('5227774fcb8b10bd2d8714f0a716a75d7a2df240a9f2a49beb76710b1c0fc619', 'https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux_ppc64le.run')}, - '10.1.243': { - 'Linux-x86_64': ('e7c22dc21278eb1b82f34a60ad7640b41ad3943d929bebda3008b72536855d31', 'https://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run'), - 'Linux-ppc64le': ('b198002eef010bab9e745ae98e47567c955d00cf34cc8f8d2f0a6feb810523bf', 'https://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux_ppc64le.run')}, - '10.0.130': { - 'Linux-x86_64': ('92351f0e4346694d0fcb4ea1539856c9eb82060c25654463bfd8574ec35ee39a', 'https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda_10.0.130_410.48_linux')}, - '9.2.88': { - 'Linux-x86_64': ('8d02cc2a82f35b456d447df463148ac4cc823891be8820948109ad6186f2667c', 'https://developer.nvidia.com/compute/cuda/9.2/Prod/local_installers/cuda_9.2.88_396.26_linux')}, - '9.1.85': { - 'Linux-x86_64': ('8496c72b16fee61889f9281449b5d633d0b358b46579175c275d85c9205fe953', 'https://developer.nvidia.com/compute/cuda/9.1/Prod/local_installers/cuda_9.1.85_387.26_linux')}, - '9.0.176': { - 'Linux-x86_64': ('96863423feaa50b5c1c5e1b9ec537ef7ba77576a3986652351ae43e66bcd080c', 'https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda_9.0.176_384.81_linux-run')}, - '8.0.61': { - 'Linux-x86_64': ('9ceca9c2397f841024e03410bfd6eabfd72b384256fbed1c1e4834b5b0ce9dc4', 'https://developer.nvidia.com/compute/cuda/8.0/Prod2/local_installers/cuda_8.0.61_375.26_linux-run')}, - '8.0.44': { - 'Linux-x86_64': ('64dc4ab867261a0d690735c46d7cc9fc60d989da0d69dc04d1714e409cacbdf0', 'https://developer.nvidia.com/compute/cuda/8.0/prod/local_installers/cuda_8.0.44_linux-run')}, - '7.5.18': { - 'Linux-x86_64': ('08411d536741075131a1858a68615b8b73c51988e616e83b835e4632eea75eec', 'https://developer.download.nvidia.com/compute/cuda/7.5/Prod/local_installers/cuda_7.5.18_linux.run')}, - '6.5.14': { - 'Linux-x86_64': ('f3e527f34f317314fe8fcd8c85f10560729069298c0f73105ba89225db69da48', 'https://developer.download.nvidia.com/compute/cuda/6_5/rel/installers/cuda_6.5.14_linux_64.run')}, - '6.0.37': { - 'Linux-x86_64': ('991e436c7a6c94ec67cf44204d136adfef87baa3ded270544fa211179779bc40', 'https://developer.download.nvidia.com/compute/cuda/6_0/rel/installers/cuda_6.0.37_linux_64.run')}, + "11.7.0": { + "Linux-aarch64": ( + "e777839a618ca9a3d5ad42ded43a1b6392af2321a7327635a4afcc986876a21b", + "https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux_sbsa.run", + ), + "Linux-x86_64": ( + "087fdfcbba1f79543b1f78e43a8dfdac5f6db242d042dde820e16dc185892f26", + "https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run", + ), + "Linux-ppc64le": ( + "74a507ac54067c258e6b7c9063c98d411116ecc5c5397b1f6e6a999e86dff08a", + "https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux_ppc64le.run", + ), + }, + "11.6.2": { + "Linux-aarch64": ( + "b20c014c6bba36b13c50da167ad42e9bd1cea24f3b6297b495ea129c0889f36e", + "https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux_sbsa.run", + ), + "Linux-x86_64": ( + "99b7a73dcc52a52cef4c1fceb4a60c3015ac9b6404082c1677d9efdaba1d4593", + "https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux.run", + ), + "Linux-ppc64le": ( + "869232ff8dbf295a71609738ac9e1b0079ca75597b427f1c026f42b36896afe8", + "https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux_ppc64le.run", + ), + }, + "11.6.1": { + "Linux-aarch64": ( + "80586b003d58030004d465f5331dc69ee26c95a29516fb2488ff10f034139cb2", + "https://developer.download.nvidia.com/compute/cuda/11.6.1/local_installers/cuda_11.6.1_510.47.03_linux_sbsa.run", + ), + "Linux-x86_64": ( + "ab219afce00b74200113269866fbff75ead037bcfc23551a8338c2684c984d7e", + "https://developer.download.nvidia.com/compute/cuda/11.6.1/local_installers/cuda_11.6.1_510.47.03_linux.run", + ), + "Linux-ppc64le": ( + "ef762efbc00b67d572823c6ec338cc2c0cf0c096f41e6bce18e8d4501f260956", + "https://developer.download.nvidia.com/compute/cuda/11.6.1/local_installers/cuda_11.6.1_510.47.03_linux_ppc64le.run", + ), + }, + "11.6.0": { + "Linux-aarch64": ( + "5898579f5e59b708520883cb161089f5e4f3426158d1e9f973c49d224085d1d2", + "https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda_11.6.0_510.39.01_linux_sbsa.run", + ), + "Linux-x86_64": ( + "1783da6d63970786040980b57fa3cb6420142159fc7d0e66f8f05c4905d98c83", + "https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda_11.6.0_510.39.01_linux.run", + ), + "Linux-ppc64le": ( + "c86b866a42baf59ddc6f1f4a79e6d77213c90749e77e574f0e0d796a749ab7d0", + "https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda_11.6.0_510.39.01_linux_ppc64le.run", + ), + }, + "11.5.2": { + "Linux-aarch64": ( + "31337c8bdc224fa1bd07bc4b6a745798392428118cc8ea0fa4446ee4ad47dd30", + "https://developer.download.nvidia.com/compute/cuda/11.5.2/local_installers/cuda_11.5.2_495.29.05_linux_sbsa.run", + ), + "Linux-x86_64": ( + "74959abf02bcba526f0a3aae322c7641b25da040ccd6236d07038f81997b73a6", + "https://developer.download.nvidia.com/compute/cuda/11.5.2/local_installers/cuda_11.5.2_495.29.05_linux.run", + ), + "Linux-ppc64le": ( + "45c468f430436b3e95d5e485a6ba0ec1fa2b23dc6c551c1307b79996ecf0a7ed", + "https://developer.download.nvidia.com/compute/cuda/11.5.2/local_installers/cuda_11.5.2_495.29.05_linux_ppc64le.run", + ), + }, + "11.5.1": { + "Linux-aarch64": ( + "73e1d0e97c7fa686efe7e00fb1e5f179372c4eec8e14d4f44ab58d5f6cf57f63", + "https://developer.download.nvidia.com/compute/cuda/11.5.1/local_installers/cuda_11.5.1_495.29.05_linux_sbsa.run", + ), + "Linux-x86_64": ( + "60bea2fc0fac95574015f865355afbf599422ec2c85554f5f052b292711a4bca", + "https://developer.download.nvidia.com/compute/cuda/11.5.1/local_installers/cuda_11.5.1_495.29.05_linux.run", + ), + "Linux-ppc64le": ( + "9e0e494d945634fe8ad3e12d7b91806aa4220ed27487bb211030d651b27c67a9", + "https://developer.download.nvidia.com/compute/cuda/11.5.1/local_installers/cuda_11.5.1_495.29.05_linux_ppc64le.run", + ), + }, + "11.5.0": { + "Linux-aarch64": ( + "6ea9d520cc956cc751a5ac54f4acc39109627f4e614dd0b1a82cc86f2aa7d8c4", + "https://developer.download.nvidia.com/compute/cuda/11.5.0/local_installers/cuda_11.5.0_495.29.05_linux_sbsa.run", + ), + "Linux-x86_64": ( + "ae0a1693d9497cf3d81e6948943e3794636900db71c98d58eefdacaf7f1a1e4c", + "https://developer.download.nvidia.com/compute/cuda/11.5.0/local_installers/cuda_11.5.0_495.29.05_linux.run", + ), + "Linux-ppc64le": ( + "95baefdc5adf165189407b119861ffb2e9800fd94d7fc81d10fb81ed36dc12db", + "https://developer.download.nvidia.com/compute/cuda/11.5.0/local_installers/cuda_11.5.0_495.29.05_linux_ppc64le.run", + ), + }, + "11.4.4": { + "Linux-aarch64": ( + "c5c08531e48e8fdc2704fa1c1f7195f2c7edd2ee10a466d0e24d05b77d109435", + "https://developer.download.nvidia.com/compute/cuda/11.4.4/local_installers/cuda_11.4.4_470.82.01_linux_sbsa.run", + ), + "Linux-x86_64": ( + "44545a7abb4b66dfc201dcad787b5e8352e5b7ddf3e3cc5b2e9177af419c25c8", + "https://developer.download.nvidia.com/compute/cuda/11.4.4/local_installers/cuda_11.4.4_470.82.01_linux.run", + ), + "Linux-ppc64le": ( + "c71cd4e6c05fde11c0485369a73e7f356080e7a18f0e3ad7244e8fc03a9dd3e2", + "https://developer.download.nvidia.com/compute/cuda/11.4.4/local_installers/cuda_11.4.4_470.82.01_linux_ppc64le.run", + ), + }, + "11.4.3": { + "Linux-aarch64": ( + "e02db34a487ea3de3eec9db80efd09f12eb69d55aca686cecaeae96a9747b1d4", + "https://developer.download.nvidia.com/compute/cuda/11.4.3/local_installers/cuda_11.4.3_470.82.01_linux_sbsa.run", + ), + "Linux-x86_64": ( + "749183821ffc051e123f12ebdeb171b263d55b86f0dd7c8f23611db1802d6c37", + "https://developer.download.nvidia.com/compute/cuda/11.4.3/local_installers/cuda_11.4.3_470.82.01_linux.run", + ), + "Linux-ppc64le": ( + "08f29cc3ed0b3b82dd9b007186237be2352bb552f99230c450a25e768f5754ee", + "https://developer.download.nvidia.com/compute/cuda/11.4.3/local_installers/cuda_11.4.3_470.82.01_linux_ppc64le.run", + ), + }, + "11.4.2": { + "Linux-aarch64": ( + "f2c4a52e06329606c8dfb7c5ea3f4cb4c0b28f9d3fdffeeb734fcc98daf580d8", + "https://developer.download.nvidia.com/compute/cuda/11.4.2/local_installers/cuda_11.4.2_470.57.02_linux_sbsa.run", + ), + "Linux-x86_64": ( + "bbd87ca0e913f837454a796367473513cddef555082e4d86ed9a38659cc81f0a", + "https://developer.download.nvidia.com/compute/cuda/11.4.2/local_installers/cuda_11.4.2_470.57.02_linux.run", + ), + "Linux-ppc64le": ( + "a917c2e53dc13fdda7def71fd40920bf3809d5a2caa3e9acfe377fb9fb22f12d", + "https://developer.download.nvidia.com/compute/cuda/11.4.2/local_installers/cuda_11.4.2_470.57.02_linux_ppc64le.run", + ), + }, + "11.4.1": { + "Linux-aarch64": ( + "8efa725a41dfd3c0c0f453c2dd535d149154102bf2b791718859417b4f84f922", + "https://developer.download.nvidia.com/compute/cuda/11.4.1/local_installers/cuda_11.4.1_470.57.02_linux_sbsa.run", + ), + "Linux-x86_64": ( + "dd6c339a719989d2518f5d54eeac1ed707d0673f8664ba0c4d4b2af7c3ba0005", + "https://developer.download.nvidia.com/compute/cuda/11.4.1/local_installers/cuda_11.4.1_470.57.02_linux.run", + ), + "Linux-ppc64le": ( + "dd92ca04f76ad938da3480e2901c0e52dbff6028ada63c09071ed9e3055dc361", + "https://developer.download.nvidia.com/compute/cuda/11.4.1/local_installers/cuda_11.4.1_470.57.02_linux_ppc64le.run", + ), + }, + "11.4.0": { + "Linux-aarch64": ( + "f0c8e80d98a601ddca031b6764459984366008c7d3847e7c7f99b36bd4438e3c", + "https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux_sbsa.run", + ), + "Linux-x86_64": ( + "d219db30f7415a115a4ea22bdbb5984b0a18f7f891cad6074c5da45d223aaa4b", + "https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux.run", + ), + "Linux-ppc64le": ( + "6eb2fd0d9d5bc39fb243b5e1789ff827f325d098cd1fbb828a0499552b9544cc", + "https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux_ppc64le.run", + ), + }, + "11.3.1": { + "Linux-aarch64": ( + "39990d3da88b21289ac20850bc183f0b66275f32e1f562b551c05843bf506e4c", + "https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux_sbsa.run", + ), + "Linux-x86_64": ( + "ad93ea98efced35855c58d3a0fc326377c60917cb3e8c017d3e6d88819bf2934", + "https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux.run", + ), + "Linux-ppc64le": ( + "220f2c10a21500d62b03c6848c1659ebb3a8e10dc0915ab87b86b397058407c5", + "https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux_ppc64le.run", + ), + }, + "11.3.0": { + "Linux-aarch64": ( + "f7b284cf055fcf18be9a9aa216e3c7968d3e011446eb0c7200a3353c223ce718", + "https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda_11.3.0_465.19.01_linux_sbsa.run", + ), + "Linux-x86_64": ( + "262da7f77db177b153a6b70b5812210f3f72f00eb608ab9cf2a4582328c4bf5c", + "https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda_11.3.0_465.19.01_linux.run", + ), + "Linux-ppc64le": ( + "c0010107933b575a87e27b1293e5dc32b74201486f4ae2f4c8695ea727d22857", + "https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda_11.3.0_465.19.01_linux_ppc64le.run", + ), + }, + "11.2.2": { + "Linux-aarch64": ( + "2f915ad631331eebdafaabd971723a60290ae8bb090d771075b9e6a0b28cbae6", + "https://developer.download.nvidia.com/compute/cuda/11.2.2/local_installers/cuda_11.2.2_460.32.03_linux_sbsa.run", + ), + "Linux-x86_64": ( + "0a2e477224af7f6003b49edfd2bfee07667a8148fe3627cfd2765f6ad72fa19d", + "https://developer.download.nvidia.com/compute/cuda/11.2.2/local_installers/cuda_11.2.2_460.32.03_linux.run", + ), + "Linux-ppc64le": ( + "2304ec235fe5d1f8bf75f00dc2c2d11473759dc23428dbbd5fb5040bc8c757e3", + "https://developer.download.nvidia.com/compute/cuda/11.2.2/local_installers/cuda_11.2.2_460.32.03_linux_ppc64le.run", + ), + }, + "11.2.1": { + "Linux-aarch64": ( + "4b322fa6477d1a2cd2f2f526fa520c0f90bef2c264ef8435cb016bebb5456c5e", + "https://developer.download.nvidia.com/compute/cuda/11.2.1/local_installers/cuda_11.2.1_460.32.03_linux_sbsa.run", + ), + "Linux-x86_64": ( + "1da98cb897cc5f58a7445a4a66ca4f6926867706cb3af58a669cdcd8dc3d17c8", + "https://developer.download.nvidia.com/compute/cuda/11.2.1/local_installers/cuda_11.2.1_460.32.03_linux.run", + ), + "Linux-ppc64le": ( + "b3e8b6cd76872deb3acd050d32e197bc1c655e142b169070f0f9753680461a3f", + "https://developer.download.nvidia.com/compute/cuda/11.2.1/local_installers/cuda_11.2.1_460.32.03_linux_ppc64le.run", + ), + }, + "11.2.0": { + "Linux-aarch64": ( + "c11dc274660e9b47b0f25ca66861a7406246a7191f1b04d0710515fcac0fa6cd", + "https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda_11.2.0_460.27.04_linux_sbsa.run", + ), + "Linux-x86_64": ( + "9c50283241ac325d3085289ed9b9c170531369de41165ce271352d4a898cbdce", + "https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda_11.2.0_460.27.04_linux.run", + ), + "Linux-ppc64le": ( + "adc3267df5dbfdaf51cb4c9b227ba6bfd979a39d9b4136bba0eba6b1dd2a2731", + "https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda_11.2.0_460.27.04_linux_ppc64le.run", + ), + }, + "11.1.1": { + "Linux-aarch64": ( + "9ab1dbafba205c06bea8c88e38cdadb3038af19cb56e7b3ba734d3d7a84b8f02", + "https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux_sbsa.run", + ), + "Linux-x86_64": ( + "3eae6727086024925ebbcef3e9a45ad379d8490768fd00f9c2d8b6fd9cd8dd8f", + "https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux.run", + ), + "Linux-ppc64le": ( + "023e571fe26ee829c98138dfc305a92279854aac7d184d255fd58c06c6af3c17", + "https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux_ppc64le.run", + ), + }, + "11.1.0": { + "Linux-aarch64": ( + "878cbd36c5897468ef28f02da50b2f546af0434a8a89d1c724a4d2013d6aa993", + "https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda_11.1.0_455.23.05_linux_sbsa.run", + ), + "Linux-x86_64": ( + "858cbab091fde94556a249b9580fadff55a46eafbcb4d4a741d2dcd358ab94a5", + "https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda_11.1.0_455.23.05_linux.run", + ), + "Linux-ppc64le": ( + "a561e6f7f659bc4100e4713523b0b8aad6b36aa77fac847f6423e7780c750064", + "https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda_11.1.0_455.23.05_linux_ppc64le.run", + ), + }, + "11.0.3": { + "Linux-aarch64": ( + "1e24f61f79c1043aa3d1d126ff6158daa03a62a51b5195a2ed5fbe75c3b718f3", + "https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux_sbsa.run", + ), + "Linux-x86_64": ( + "b079c4e408adf88c3f1ffb8418a97dc4227c37935676b4bf4ca0beec6c328cc0", + "https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux.run", + ), + "Linux-ppc64le": ( + "4775b21df004b1433bafff9b48a324075c008509f4c0fe28cd060d042d2e0794", + "https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux_ppc64le.run", + ), + }, + "11.0.2": { + "Linux-aarch64": ( + "23851e30f7c47a1baad92891abde0adbc783de5962c7480b9725198ceacda4a0", + "https://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux_sbsa.run", + ), + "Linux-x86_64": ( + "48247ada0e3f106051029ae8f70fbd0c238040f58b0880e55026374a959a69c1", + "https://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run", + ), + "Linux-ppc64le": ( + "db06d0f3fbf6f7aa1f106fc921ad1c86162210a26e8cb65b171c5240a3bf75da", + "https://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux_ppc64le.run", + ), + }, + "10.2.89": { + "Linux-x86_64": ( + "560d07fdcf4a46717f2242948cd4f92c5f9b6fc7eae10dd996614da913d5ca11", + "https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run", + ), + "Linux-ppc64le": ( + "5227774fcb8b10bd2d8714f0a716a75d7a2df240a9f2a49beb76710b1c0fc619", + "https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux_ppc64le.run", + ), + }, + "10.1.243": { + "Linux-x86_64": ( + "e7c22dc21278eb1b82f34a60ad7640b41ad3943d929bebda3008b72536855d31", + "https://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run", + ), + "Linux-ppc64le": ( + "b198002eef010bab9e745ae98e47567c955d00cf34cc8f8d2f0a6feb810523bf", + "https://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux_ppc64le.run", + ), + }, + "10.0.130": { + "Linux-x86_64": ( + "92351f0e4346694d0fcb4ea1539856c9eb82060c25654463bfd8574ec35ee39a", + "https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda_10.0.130_410.48_linux", + ) + }, + "9.2.88": { + "Linux-x86_64": ( + "8d02cc2a82f35b456d447df463148ac4cc823891be8820948109ad6186f2667c", + "https://developer.nvidia.com/compute/cuda/9.2/Prod/local_installers/cuda_9.2.88_396.26_linux", + ) + }, + "9.1.85": { + "Linux-x86_64": ( + "8496c72b16fee61889f9281449b5d633d0b358b46579175c275d85c9205fe953", + "https://developer.nvidia.com/compute/cuda/9.1/Prod/local_installers/cuda_9.1.85_387.26_linux", + ) + }, + "9.0.176": { + "Linux-x86_64": ( + "96863423feaa50b5c1c5e1b9ec537ef7ba77576a3986652351ae43e66bcd080c", + "https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda_9.0.176_384.81_linux-run", + ) + }, + "8.0.61": { + "Linux-x86_64": ( + "9ceca9c2397f841024e03410bfd6eabfd72b384256fbed1c1e4834b5b0ce9dc4", + "https://developer.nvidia.com/compute/cuda/8.0/Prod2/local_installers/cuda_8.0.61_375.26_linux-run", + ) + }, + "8.0.44": { + "Linux-x86_64": ( + "64dc4ab867261a0d690735c46d7cc9fc60d989da0d69dc04d1714e409cacbdf0", + "https://developer.nvidia.com/compute/cuda/8.0/prod/local_installers/cuda_8.0.44_linux-run", + ) + }, + "7.5.18": { + "Linux-x86_64": ( + "08411d536741075131a1858a68615b8b73c51988e616e83b835e4632eea75eec", + "https://developer.download.nvidia.com/compute/cuda/7.5/Prod/local_installers/cuda_7.5.18_linux.run", + ) + }, + "6.5.14": { + "Linux-x86_64": ( + "f3e527f34f317314fe8fcd8c85f10560729069298c0f73105ba89225db69da48", + "https://developer.download.nvidia.com/compute/cuda/6_5/rel/installers/cuda_6.5.14_linux_64.run", + ) + }, + "6.0.37": { + "Linux-x86_64": ( + "991e436c7a6c94ec67cf44204d136adfef87baa3ded270544fa211179779bc40", + "https://developer.download.nvidia.com/compute/cuda/6_0/rel/installers/cuda_6.0.37_linux_64.run", + ) + }, } @@ -146,8 +406,8 @@ class Cuda(Package): homepage = "https://developer.nvidia.com/cuda-zone" - maintainers = ['ax3l', 'Rombur'] - executables = ['^nvcc$'] + maintainers = ["ax3l", "Rombur"] + executables = ["^nvcc$"] for ver, packages in _versions.items(): key = "{0}-{1}".format(platform.system(), platform.machine()) @@ -164,55 +424,62 @@ class Cuda(Package): # macOS NVIDIA drivers at # https://www.nvidia.com/en-us/drivers/cuda/mac-driver-archive/ mention # Mojave support -- only macOS High Sierra 10.13 is supported. - conflicts('arch=darwin-mojave-x86_64') - - variant('dev', default=False, description='Enable development dependencies, i.e to use cuda-gdb') - variant('allow-unsupported-compilers', default=False, sticky=True, - description='Allow unsupported host compiler and CUDA version combinations') - - depends_on('libxml2', when='@10.1.243:') + conflicts("arch=darwin-mojave-x86_64") + + variant( + "dev", default=False, description="Enable development dependencies, i.e to use cuda-gdb" + ) + variant( + "allow-unsupported-compilers", + default=False, + sticky=True, + description="Allow unsupported host compiler and CUDA version combinations", + ) + + depends_on("libxml2", when="@10.1.243:") # cuda-gdb needed libncurses.so.5 before 11.4.0 # see https://docs.nvidia.com/cuda/archive/11.3.1/cuda-gdb/index.html#common-issues-oss # see https://docs.nvidia.com/cuda/archive/11.4.0/cuda-gdb/index.html#release-notes - depends_on('ncurses abi=5', type='run', when='@:11.3.99+dev') + depends_on("ncurses abi=5", type="run", when="@:11.3.99+dev") - provides('opencl@:1.2', when='@7:') - provides('opencl@:1.1', when='@:6') + provides("opencl@:1.2", when="@7:") + provides("opencl@:1.1", when="@:6") @classmethod def determine_version(cls, exe): - output = Executable(exe)('--version', output=str, error=str) - match = re.search(r'Cuda compilation tools, release .*?, V(\S+)', - output) + output = Executable(exe)("--version", output=str, error=str) + match = re.search(r"Cuda compilation tools, release .*?, V(\S+)", output) return match.group(1) if match else None def setup_build_environment(self, env): - if self.spec.satisfies('@:8.0.61'): + if self.spec.satisfies("@:8.0.61"): # Perl 5.26 removed current directory from module search path, # CUDA 9 has a fix for this, but CUDA 8 and lower don't. - env.append_path('PERL5LIB', self.stage.source_path) + env.append_path("PERL5LIB", self.stage.source_path) - if self.spec.satisfies('@10.1.243:'): - libxml2_home = self.spec['libxml2'].prefix - env.set('LIBXML2HOME', libxml2_home) - env.append_path('LD_LIBRARY_PATH', libxml2_home.lib) + if self.spec.satisfies("@10.1.243:"): + libxml2_home = self.spec["libxml2"].prefix + env.set("LIBXML2HOME", libxml2_home) + env.append_path("LD_LIBRARY_PATH", libxml2_home.lib) def setup_dependent_build_environment(self, env, dependent_spec): - env.set('CUDAHOSTCXX', dependent_spec.package.compiler.cxx) + env.set("CUDAHOSTCXX", dependent_spec.package.compiler.cxx) def setup_run_environment(self, env): - env.set('CUDA_HOME', self.prefix) + env.set("CUDA_HOME", self.prefix) def install(self, spec, prefix): - if os.path.exists('/tmp/cuda-installer.log'): + if os.path.exists("/tmp/cuda-installer.log"): try: - os.remove('/tmp/cuda-installer.log') + os.remove("/tmp/cuda-installer.log") except OSError: - if spec.satisfies('@10.1:'): - tty.die("The cuda installer will segfault due to the " - "presence of /tmp/cuda-installer.log " - "please remove the file and try again ") - runfile = glob(join_path(self.stage.source_path, 'cuda*_linux*'))[0] + if spec.satisfies("@10.1:"): + tty.die( + "The cuda installer will segfault due to the " + "presence of /tmp/cuda-installer.log " + "please remove the file and try again " + ) + runfile = glob(join_path(self.stage.source_path, "cuda*_linux*"))[0] # Note: NVIDIA does not officially support many newer versions of # compilers. For example, on CentOS 6, you must use GCC 4.4.7 or @@ -224,60 +491,59 @@ class Cuda(Package): # CUDA 10.1 on ppc64le fails to copy some files, the workaround is adapted from # https://forums.developer.nvidia.com/t/cuda-10-1-243-10-1-update-2-ppc64le-run-file-installation-issue/82433 # See also #21170 - if spec.satisfies('@10.1.243') and platform.machine() == 'ppc64le': + if spec.satisfies("@10.1.243") and platform.machine() == "ppc64le": includedir = "targets/ppc64le-linux/include" os.makedirs(os.path.join(prefix, includedir)) os.makedirs(os.path.join(prefix, "src")) os.symlink(includedir, os.path.join(prefix, "include")) - install_shell = which('sh') + install_shell = which("sh") - if self.spec.satisfies('@:8.0.61'): + if self.spec.satisfies("@:8.0.61"): # Perl 5.26 removed current directory from module search path. # We are addressing this by exporting `PERL5LIB` earlier, but for some # reason, it is not enough. One more file needs to be extracted before # running the actual installer. This solution is one of the commonly # found on the Internet, when people try to install CUDA <= 8 manually. # For example: https://askubuntu.com/a/1087842 - arguments = [runfile, '--tar', 'mxvf', './InstallUtils.pm'] + arguments = [runfile, "--tar", "mxvf", "./InstallUtils.pm"] install_shell(*arguments) # CUDA 10.1+ has different cmdline options for the installer arguments = [ - runfile, # the install script - '--silent', # disable interactive prompts - '--override', # override compiler version checks - '--toolkit', # install CUDA Toolkit + runfile, # the install script + "--silent", # disable interactive prompts + "--override", # override compiler version checks + "--toolkit", # install CUDA Toolkit ] - if spec.satisfies('@7:'): + if spec.satisfies("@7:"): # use stage dir instead of /tmp - mkdir(join_path(self.stage.path, 'tmp')) - arguments.append('--tmpdir=%s' % join_path(self.stage.path, 'tmp')) + mkdir(join_path(self.stage.path, "tmp")) + arguments.append("--tmpdir=%s" % join_path(self.stage.path, "tmp")) - if spec.satisfies('@10.1:'): - arguments.append('--installpath=%s' % prefix) # Where to install + if spec.satisfies("@10.1:"): + arguments.append("--installpath=%s" % prefix) # Where to install else: - arguments.append('--verbose') # Verbose log file - arguments.append('--toolkitpath=%s' % prefix) # Where to install + arguments.append("--verbose") # Verbose log file + arguments.append("--toolkitpath=%s" % prefix) # Where to install install_shell(*arguments) try: - os.remove('/tmp/cuda-installer.log') + os.remove("/tmp/cuda-installer.log") except OSError: pass @property def libs(self): - libs = find_libraries('libcudart', root=self.prefix, shared=True, - recursive=True) + libs = find_libraries("libcudart", root=self.prefix, shared=True, recursive=True) filtered_libs = [] # CUDA 10.0 provides Compatability libraries for running newer versions # of CUDA with older drivers. These do not work with newer drivers. for lib in libs: parts = lib.split(os.sep) - if 'compat' not in parts and 'stubs' not in parts: + if "compat" not in parts and "stubs" not in parts: filtered_libs.append(lib) return LibraryList(filtered_libs) |