Merge pull request #15 from jjhelmus/micro_ver

add micro version to cudatoolkit 10.0 package
numba · Feb 8, 2019 · 9827d76 · 9827d76
2 parents a48b39c + a37525e
commit 9827d76
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/condarecipe10.0/meta.yaml b/condarecipe10.0/meta.yaml
@@ -1,6 +1,7 @@
 package:
    name: cudatoolkit
-   version: 10.0
+   # match the package version to the libcudart.so version
+   version: 10.0.130
 
 build:
   number: 0

diff --git a/scripts/build.py b/scripts/build.py
@@ -755,6 +755,10 @@ def _main():
 
     # package version decl must match cuda release version
     cu_version = os.environ['PKG_VERSION']
+    # keep only the major.minor version (10.0) if micro (10.0.130) is present
+    major_minor, micro = cu_version.rsplit('.', 1)
+    if '.' in major_minor:
+        cu_version = major_minor
 
     # get an extractor
     plat = getplatform()