68 lines
2.3 KiB
YAML
68 lines
2.3 KiB
YAML
name: 'cuda-toolkit'
|
|
description: 'Installs NVIDIA CUDA Toolkit and adds it to PATH'
|
|
author: 'Jim Verheijde'
|
|
inputs:
|
|
cuda:
|
|
description: 'Cuda version'
|
|
required: false
|
|
default: '12.9.1'
|
|
sub-packages:
|
|
description:
|
|
'Only installs specified subpackages, must be in the form of a JSON array.
|
|
For example, if you only want to install nvcc and visual studio
|
|
integration: ["nvcc", "visual_studio_integration"] double quotes required!
|
|
Note that if you want to use this on Linux, ''network'' method MUST be
|
|
used.'
|
|
required: false
|
|
default: '[]'
|
|
non-cuda-sub-packages:
|
|
description:
|
|
'Only installs specified subpackages that do not have the cuda prefix,
|
|
must be in the form of a JSON array. For example, if you only want to
|
|
install libcublas and libcufft: ["libcublas", "libcufft"] double quotes
|
|
required! Note that this only works with ''network'' method on only on
|
|
Linux.'
|
|
required: false
|
|
default: '[]'
|
|
method:
|
|
description:
|
|
"Installation method, can be either 'local' or 'network'. 'local'
|
|
downloads the entire installer with all packages and runs that (you can
|
|
still only install certain packages with sub-packages on Windows).
|
|
'network' downloads a smaller executable which only downloads necessary
|
|
packages which you can define in subPackages"
|
|
required: false
|
|
default: 'local'
|
|
linux-local-args:
|
|
description:
|
|
'(Linux and ''local'' method only) override arguments for the linux .run
|
|
installer. For example if you don''t want samples use ["--toolkit"] double
|
|
quotes required!'
|
|
required: false
|
|
default: '["--toolkit", "--samples"]'
|
|
use-github-cache:
|
|
description:
|
|
'Use GitHub cache to cache downloaded installer on GitHub servers'
|
|
required: false
|
|
default: 'true'
|
|
use-local-cache:
|
|
description:
|
|
'Use local cache to cache downloaded installer on the local runner'
|
|
required: false
|
|
default: 'true'
|
|
log-file-suffix:
|
|
description: 'Suffix of log file name in artifact'
|
|
required: false
|
|
default: 'log.txt'
|
|
outputs:
|
|
CUDA_PATH:
|
|
description: 'Path to CUDA installation'
|
|
cuda:
|
|
description: 'Version of CUDA installed'
|
|
runs:
|
|
using: 'node20'
|
|
main: dist/index.js
|
|
branding:
|
|
icon: box
|
|
color: green
|