get-cuda

Automatically generated README for this automation recipe: get-cuda

Category: CUDA automation

License: Apache 2.0

Notes from the authors, contributors and users: README-extra

System dependencies

Download CUDA toolkit.
Download cuDNN.
Download TensorRT.
CM meta description for this script: _cm.yaml
Output cached? True

Reuse this script in your project

Install MLCommons CM automation meta-framework

Pull CM repository with this automation recipe (CM script)

cm pull repo mlcommons@cm4mlops

Print CM help from the command line

cmr "get cuda cuda-compiler cuda-lib toolkit lib nvcc get-nvcc get-cuda 46d133d9ef92422d" --help

Run this script

CLICLI AltPythonDocker

Run this script via CLI

cm run script --tags=get,cuda,cuda-compiler,cuda-lib,toolkit,lib,nvcc,get-nvcc,get-cuda,46d133d9ef92422d[,variations] [--input_flags]

Run this script via CLI (alternative)

cmr "get cuda cuda-compiler cuda-lib toolkit lib nvcc get-nvcc get-cuda 46d133d9ef92422d [variations]" [--input_flags]

Run this script from Python

import cmind

r = cmind.access({'action':'run'
              'automation':'script',
              'tags':'get,cuda,cuda-compiler,cuda-lib,toolkit,lib,nvcc,get-nvcc,get-cuda,46d133d9ef92422d'
              'out':'con',
              ...
              (other input keys for this script)
              ...
             })

if r['return']>0:
    print (r['error'])

Run this script via Docker (beta)

cm docker script "get cuda cuda-compiler cuda-lib toolkit lib nvcc get-nvcc get-cuda 46d133d9ef92422d[variations]" [--input_flags]

VariationsInput Flag MappingDefault environment

Variations

No group (any combination of variations can be selected)
Click here to expand this section.
- _cudnn
  - ENV variables:
    - CM_CUDA_NEEDS_CUDNN: yes
- _package-manager
  - ENV variables:
    - CM_CUDA_PACKAGE_MANAGER_INSTALL: yes
Group "installation-mode"
Click here to expand this section.
- _lib-only
  - ENV variables:
    - CM_CUDA_FULL_TOOLKIT_INSTALL: no
    - CM_TMP_FILE_TO_CHECK_UNIX: libcudart.so
    - CM_TMP_FILE_TO_CHECK_WINDOWS: libcudart.dll
- _toolkit (default)
  - ENV variables:
    - CM_CUDA_FULL_TOOLKIT_INSTALL: yes
    - CM_TMP_FILE_TO_CHECK_UNIX: nvcc
    - CM_TMP_FILE_TO_CHECK_WINDOWS: nvcc.exe

Default variations

_toolkit

Script flags mapped to environment

--cudnn_tar_file=value → CM_CUDNN_TAR_FILE_PATH=value
--cudnn_tar_path=value → CM_CUDNN_TAR_FILE_PATH=value

Default environment

These keys can be updated via --env.KEY=VALUE or env dictionary in @input.json or using script flags.

CM_CUDA_PATH_LIB_CUDNN_EXISTS: no
CM_REQUIRE_INSTALL: no

Native script being run

Linux/macOSWindows

run.sh

run.bat

Script output

cmr "get cuda cuda-compiler cuda-lib toolkit lib nvcc get-nvcc get-cuda 46d133d9ef92422d [variations]" [--input_flags] -j