Merge pull request #21952 from adibbley/cuda-compat

Adding cuda-compat recipe
conda-forge · Jun 23, 2023 · d9c104c · d9c104c
2 parents 9319d9b + ce2c174
commit d9c104c
Show file tree

Hide file tree

Showing 2 changed files with 107 additions and 0 deletions.
diff --git a/recipes/cuda-compat/build.sh b/recipes/cuda-compat/build.sh
@@ -0,0 +1,16 @@
+#!/bin/bash
+
+# Install to conda style directories
+[[ -d lib64 ]] && mv lib64 lib
+mkdir -p ${PREFIX}/cuda-compat
+
+cp -v lib/libcuda.so.${DRV_VERSION} ${PREFIX}/cuda-compat
+ln -sv libcuda.so.${DRV_VERSION} ${PREFIX}/cuda-compat/libcuda.so.1
+ln -sv libcuda.so.1 ${PREFIX}/cuda-compat/libcuda.so
+cp -v lib/libnvidia-nvvm.so.${DRV_VERSION} ${PREFIX}/cuda-compat
+ln -sv libnvidia-nvvm.so.${DRV_VERSION} ${PREFIX}/cuda-compat/libnvidia-nvvm.so.4
+ln -sv libnvidia-nvvm.so.4 ${PREFIX}/cuda-compat/libnvidia-nvvm.so
+cp -v lib/libnvidia-ptxjitcompiler.so.${DRV_VERSION} ${PREFIX}/cuda-compat
+ln -sv libnvidia-ptxjitcompiler.so.${DRV_VERSION} ${PREFIX}/cuda-compat/libnvidia-ptxjitcompiler.so.1
+cp -v lib/libcudadebugger.so.${DRV_VERSION} ${PREFIX}/cuda-compat
+ln -sv libcudadebugger.so.${DRV_VERSION} ${PREFIX}/cuda-compat/libcudadebugger.so.1
diff --git a/recipes/cuda-compat/meta.yaml b/recipes/cuda-compat/meta.yaml
@@ -0,0 +1,91 @@
+{% set name = "cuda-compat" %}
+{% set driver_version = "525.60.13" %}
+{% set cuda_version = "12.0.0" %}
+{% set platform = "linux-x86_64" %}  # [linux64]
+{% set platform = "linux-ppc64le" %}  # [ppc64le]
+{% set platform = "linux-sbsa" %}  # [aarch64]
+{% set extension = "tar.xz" %}  # [not win]
+
+package:
+  name: {{ name|lower }}
+  version: {{ cuda_version }}
+
+source:
+  url: https://developer.download.nvidia.com/compute/cuda/redist/nvidia_driver/{{ platform }}/nvidia_driver-{{ platform }}-{{ driver_version }}-archive.{{ extension }}
+  sha256: a42691822f6e87060da5682a8e0662c86d1b20fbf31d12cb2969b819995e4138  # [linux64]
+  sha256: 4ad08f8431195580349f4984fae2d93697d2f5572837c1e7ae59c37feab291c8  # [ppc64le]
+  sha256: 1708ae3ba857559cbfc4e6c06267ed6126eff2c68fa5321a95906f50e984148c  # [aarch64]
+
+build:
+  number: 0
+  skip: true  # [osx or win]
+  script_env:
+    - DRV_VERSION={{ driver_version }}
+
+outputs:
+  - name: cuda-compat
+    requirements:
+      run:
+        - {{ pin_subpackage("cuda-compat-impl", exact=True) }}
+    test:
+      commands:
+        - exit 0
+    about:
+      home: https://developer.nvidia.com/cuda-toolkit
+      license_file: LICENSE
+      license: LicenseRef-NVIDIA-End-User-License-Agreement
+      license_url: https://docs.nvidia.com/cuda/eula/index.html
+      summary: CUDA Compatibility Platform
+      description: |
+          Package containing all the necessary CUDA driver files related to forward
+          compatibility. This compatibility enables newer CUDA runtimes to work with
+          older display drivers. Used for Tesla cards only.
+      doc_url: https://docs.nvidia.com/cuda/index.html
+
+  - name: cuda-compat-impl
+    version: {{ driver_version }}
+    files:
+      - cuda-compat
+    requirements:
+      build:
+        - {{ compiler("c") }}
+        - {{ compiler("cxx") }}
+    test:
+      commands:
+        - test -f $PREFIX/cuda-compat/libcuda.so.{{ driver_version }}
+        - test -L $PREFIX/cuda-compat/libcuda.so.1
+        - test -L $PREFIX/cuda-compat/libcuda.so
+        - test -f $PREFIX/cuda-compat/libnvidia-nvvm.so.{{ driver_version }}
+        - test -L $PREFIX/cuda-compat/libnvidia-nvvm.so.4
+        - test -L $PREFIX/cuda-compat/libnvidia-nvvm.so
+        - test -f $PREFIX/cuda-compat/libnvidia-ptxjitcompiler.so.{{ driver_version }}
+        - test -L $PREFIX/cuda-compat/libnvidia-ptxjitcompiler.so.1
+        - test -f $PREFIX/cuda-compat/libcudadebugger.so.{{ driver_version }}
+        - test -L $PREFIX/cuda-compat/libcudadebugger.so.1
+    about:
+      home: https://developer.nvidia.com/cuda-toolkit
+      license_file: LICENSE
+      license: LicenseRef-NVIDIA-End-User-License-Agreement
+      license_url: https://docs.nvidia.com/cuda/eula/index.html
+      summary: CUDA Compatibility Platform
+      description: |
+          Package containing all the necessary CUDA driver files related to forward
+          compatibility. This compatibility enables newer CUDA runtimes to work with
+          older display drivers. Used for Tesla cards only.
+      doc_url: https://docs.nvidia.com/cuda/index.html
+
+about:
+  home: https://developer.nvidia.com/cuda-toolkit
+  license_file: LICENSE
+  license: LicenseRef-NVIDIA-End-User-License-Agreement
+  license_url: https://docs.nvidia.com/cuda/eula/index.html
+  summary: CUDA Compatibility Platform
+  description: |
+      Package containing all the necessary CUDA driver files related to forward
+      compatibility. This compatibility enables newer CUDA runtimes to work with
+      older display drivers. Used for Tesla cards only.
+  doc_url: https://docs.nvidia.com/cuda/index.html
+
+extra:
+  recipe-maintainers:
+    - adibbley