-
Notifications
You must be signed in to change notification settings - Fork 57
/
action.yml
42 lines (42 loc) · 2.03 KB
/
action.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
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.6.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'
runs:
using: 'node20'
main: dist/index.js
branding:
icon: box
color: green