diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml index b07bc4f93..05e9309b2 100644 --- a/conda/environments/all_cuda-118_arch-x86_64.yaml +++ b/conda/environments/all_cuda-118_arch-x86_64.yaml @@ -9,6 +9,7 @@ dependencies: - c-compiler - clang-tools=16.0.6 - cmake>=3.26.4,!=3.30.0 +- cuda-python>=11.8.5,<12.0a0 - cuda-version=11.8 - cudatoolkit - cudf==25.2.*,>=0.0.0a0 diff --git a/conda/environments/all_cuda-125_arch-x86_64.yaml b/conda/environments/all_cuda-125_arch-x86_64.yaml index a9cd79d98..b4a928b75 100644 --- a/conda/environments/all_cuda-125_arch-x86_64.yaml +++ b/conda/environments/all_cuda-125_arch-x86_64.yaml @@ -13,6 +13,7 @@ dependencies: - cuda-cupti-dev - cuda-nvcc - cuda-nvrtc-dev +- cuda-python>=12.6.2,<13.0a0 - cuda-version=12.5 - cudf==25.2.*,>=0.0.0a0 - cuml==25.2.*,>=0.0.0a0 diff --git a/dependencies.yaml b/dependencies.yaml index 269687def..c859be47a 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -370,6 +370,16 @@ dependencies: - output_types: [conda] packages: - curl + specific: + - output_types: [conda, requirements, pyproject] + matrices: + - matrix: + cuda: "12.*" + packages: + - cuda-python>=12.6.2,<13.0a0 + - matrix: # All CUDA 11 versions + packages: + - cuda-python>=11.8.5,<12.0a0 # TODO: Remove geopandas.dataset usage in cuspatial_api_examples.ipynb test_notebooks: common: diff --git a/notebooks/cuproj_benchmark.ipynb b/notebooks/cuproj_benchmark.ipynb index 2ff673422..f5df9ab2c 100644 --- a/notebooks/cuproj_benchmark.ipynb +++ b/notebooks/cuproj_benchmark.ipynb @@ -21,7 +21,7 @@ "source": [ "import numpy as np\n", "import cupy as cp\n", - "from cuda import cudart\n", + "from cuda.bindings import runtime\n", "\n", "from pyproj import Transformer\n", "from cuproj import Transformer as cuTransformer\n", @@ -85,8 +85,8 @@ " return t.autorange()\n", "\n", "def cu_run_and_sync(transformer, grid):\n", - " transformer.transform(*grid)\n", - " cudart.cudaDeviceSynchronize()\n", + " transformer.transform(*grid)\n", + " runtime.cudaDeviceSynchronize()\n", "\n", "def run_cuproj(num_points):\n", " grid_side = int(np.sqrt(num_points))\n",