commit 631f343e5c4aaf4f72af6a67ae5d0bcdffaad484 · anil.recoil.org/opam-repository

+58

packages/conf-cuda-config/conf-cuda-config.1/opam

···

       1
       1
       +
       opam-version: "2.0"

     

       2
       2
       +
       maintainer: "Lukasz Stafiniak <lukstafi@gmail.com>"

     

       3
       3
       +
       authors: [

     

       4
       4
       +
           "Lukasz Stafiniak <lukstafi@gmail.com>"

     

       5
       5
       +
       ]

     

       6
       6
       +
       homepage: "https://docs.nvidia.com/cuda/"

     

       7
       7
       +
       license: "MIT"

     

       8
       8
       +
       build: [

     

       9
       9
       +
         ["sh" "-exc" """

     

       10
       10
       +
           if [ -z "$CUDA_PATH" ]; then

     

       11
       11
       +
             if [ -d "/usr/local/cuda" ]; then

     

       12
       12
       +
               CUDA_PATH="/usr/local/cuda"

     

       13
       13
       +
               CUDA_PREINSTALLED="true"

     

       14
       14
       +
             elif [ -d "/opt/cuda" ]; then

     

       15
       15
       +
               CUDA_PATH="/opt/cuda"

     

       16
       16
       +
               CUDA_PREINSTALLED="true"

     

       17
       17
       +
             elif [ -d "/usr/cuda" ]; then

     

       18
       18
       +
               CUDA_PATH="/usr/cuda"

     

       19
       19
       +
               CUDA_PREINSTALLED="true"

     

       20
       20
       +
             elif [ "arch" = %{os-family}% ]; then

     

       21
       21
       +
               CUDA_PATH="/opt/cuda"

     

       22
       22
       +
               CUDA_PREINSTALLED="false"

     

       23
       23
       +
             else

     

       24
       24
       +
               CUDA_PATH="/usr/local/cuda"

     

       25
       25
       +
               CUDA_PREINSTALLED="false"

     

       26
       26
       +
             fi

     

       27
       27
       +
           else

     

       28
       28
       +
             if [ -d "$CUDA_PATH" ]; then

     

       29
       29
       +
               CUDA_PREINSTALLED="true"

     

       30
       30
       +
             else

     

       31
       31
       +
               CUDA_PREINSTALLED="false"

     

       32
       32
       +
             fi

     

       33
       33
       +
           fi

     

       34
       34
       +
           if [ -z "$WSL_DISTRO_NAME" ]; then

     

       35
       35
       +
             IS_WSL="false"

     

       36
       36
       +
           else

     

       37
       37
       +
             IS_WSL="true"

     

       38
       38
       +
           fi

     

       39
       39
       +
           cat <<EOF > conf-cuda-config.config

     

       40
       40
       +
       opam-version: "2.0"

     

       41
       41
       +
       variables {

     

       42
       42
       +
         cuda_preinstalled: "$CUDA_PREINSTALLED"

     

       43
       43
       +
         cuda_path: "$CUDA_PATH"

     

       44
       44
       +
         is_wsl: "$IS_WSL"

     

       45
       45
       +
         wsl_distro_name: "$WSL_DISTRO_NAME"

     

       46
       46
       +
       }

     

       47
       47
       +
       EOF

     

       48
       48
       +
           """]

     

       49
       49
       +
       ]

     

       50
       50
       +
       post-messages: [

     

       51
       51
       +
         "NOTE: assuming CUDA will be installed under %{conf-cuda-config:cuda_path}%"

     

       52
       52
       +
         {!conf-cuda-config:cuda_preinstalled}

     

       53
       53
       +
       ]

     

       54
       54
       +
       synopsis: "Preparatory steps for the conf-cuda package"

     

       55
       55
       +
       description:

     

       56
       56
       +
         "This package checks if CUDA is already installed on the system, and if the OS runs under WSL."

     

       57
       57
       +
       bug-reports: "https://github.com/ocaml/opam-repository/issues"

     

       58
       58
       +
       flags: conf

+48

packages/conf-cuda/conf-cuda.1/opam

···

       1
       1
       +
       opam-version: "2.0"

     

       2
       2
       +
       maintainer: "Lukasz Stafiniak <lukstafi@gmail.com>"

     

       3
       3
       +
       authors: [

     

       4
       4
       +
           "NVidia"

     

       5
       5
       +
       ]

     

       6
       6
       +
       homepage: "https://docs.nvidia.com/cuda/"

     

       7
       7
       +
       license: "CUDA Toolkit End User License Agreement <https://docs.nvidia.com/cuda/eula/index.html>"

     

       8
       8
       +
       # We do not restrict availability to let users install CUDA themselves.

     

       9
       9
       +
       build: [

     

       10
       10
       +
         ["sh" "-exc" "cat <<EOF > test.c

     

       11
       11
       +
          #include  \"cuda.h\"

     

       12
       12
       +
          #include  \"nvrtc.h\"

     

       13
       13
       +
          "]

     

       14
       14
       +
         ["sh" "-exc" "cc -c $CFLAGS -I%{conf-cuda-config:cuda_path}%/include test.c"]

     

       15
       15
       +
       ]

     

       16
       16
       +
       # Note: the minimal set of packages directly from NVidia repositories, e.g. for cudajit:

     

       17
       17
       +
       # "cuda-cudart-X-Y" "cuda-cudart-dev-X-Y" "cuda-runtime-X-Y" "cuda-nvrtc-X-Y" "cuda-nvrtc-dev-X-Y"

     

       18
       18
       +
       # where e.g. X=12, Y=4 is the CUDA version.

     

       19
       19
       +
       depends: [ "conf-cuda-config" ]

     

       20
       20
       +
       setenv: [

     

       21
       21
       +
         [ CUDA_PATH = "%{conf-cuda-config:cuda_path}%" ]

     

       22
       22
       +
       ]

     

       23
       23
       +
       post-messages: [

     

       24
       24
       +
         "Make sure that CUDA is properly installed under the path %{conf-cuda-config:cuda_path}%; or properly set the CUDA_PATH environment variable and re-install opam package conf-cuda-config. See: https://docs.nvidia.com/cuda/ Failing configuration: CUDA_PATH=%{conf-cuda-config:cuda_path}%, is-WSL %{conf-cuda-config:is_wsl}%, OS family %{os-family}%"

     

       25
       25
       +
         {failure & conf-cuda-config:cuda_preinstalled}

     

       26
       26
       +
         "Execute: $ sudo add-apt-repository multiverse; sudo apt update; sudo apt-get install nvidia-cuda-dev nvidia-cuda-toolkit"

     

       27
       27
       +
         {failure & !conf-cuda-config:cuda_preinstalled & !conf-cuda-config:is_wsl & os-family = "debian" & os-distribution != "ubuntu"}

     

       28
       28
       +
         "Execute: $ sudo apt-get install nvidia-cuda-dev nvidia-cuda-toolkit"

     

       29
       29
       +
         {failure & !conf-cuda-config:cuda_preinstalled & !conf-cuda-config:is_wsl & os-distribution = "ubuntu"}

     

       30
       30
       +
         "Under WSL, manually install CUDA. See https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#network-repo-installation-for-wsl"

     

       31
       31
       +
         {failure & !conf-cuda-config:cuda_preinstalled & conf-cuda-config:is_wsl}

     

       32
       32
       +
         "Execute: $ sudo zypper in CUDA CUDA-tools"

     

       33
       33
       +
         {failure & !conf-cuda-config:is_wsl & !conf-cuda-config:cuda_preinstalled &

     

       34
       34
       +
          (os-family = "opensuse" | os-distribution = "opensuse-leap" | os-distribution = "opensuse-tumbleweed")}

     

       35
       35
       +
         "Execute: $ sudo pacman -Sy cuda"

     

       36
       36
       +
          {failure & !conf-cuda-config:is_wsl & !conf-cuda-config:cuda_preinstalled & os-family = "arch"}

     

       37
       37
       +
          "Alternatively: manually install CUDA or verify that CUDA_PATH is set properly, and re-install opam package conf-cuda-config. See: https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html"

     

       38
       38
       +
          {failure & !conf-cuda-config:is_wsl & !conf-cuda-config:cuda_preinstalled & os = "linux"}

     

       39
       39
       +
          "Manually install CUDA or verify that CUDA_PATH is set properly, and re-install opam package conf-cuda-config. See: https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html"

     

       40
       40
       +
          {failure & !conf-cuda-config:is_wsl & !conf-cuda-config:cuda_preinstalled & os = "windows"}

     

       41
       41
       +
          "If CUDA is supported on your system, manually install CUDA or verify that CUDA_PATH is set properly, and re-install opam package conf-cuda-config. See: https://docs.nvidia.com/cuda/"

     

       42
       42
       +
          {failure & !conf-cuda-config:is_wsl & !conf-cuda-config:cuda_preinstalled & os != "linux" & os != "windows"}

     

       43
       43
       +
       ]

     

       44
       44
       +
       synopsis: "Virtual package relying on a CUDA system installation"

     

       45
       45
       +
       description:

     

       46
       46
       +
         "This package can only install if CUDA is installed on the system."

     

       47
       47
       +
       bug-reports: "https://github.com/ocaml/opam-repository/issues"

     

       48
       48
       +
       flags: conf