Update with runtime dependencies and testable imports
{% set name = "cuda-python" %}
-{% set version = "11.4.0" %}
+{% set major_version = "11" %}
+{% set minor_version = "4" %}
+{% set patch_version = "0" %}
+{% set version = major_version ~ "." ~ minor_version ~ "." ~ patch_version %}
package:
name: {{ name|lower }}
@@ -28,10 +31,15 @@ requirements:
- pip
run:
- python
+ - cudatoolkit >={{ major_version ~ ".0.0" }}
+ run_constrained:
+ - __cuda >={{ major_version ~ ".0.0" }}
test:
imports:
- cuda
+ - cuda.cuda
+ - cuda.cudart
about:
home: https://github.com/NVIDIA/cuda-python
