cuda-toolkit
This action installs the NVIDIA® CUDA® Toolkit on the system. It adds the cuda install location as CUDA_PATH to GITHUB_ENV so you can access the CUDA install location in subsequent steps. CUDA_PATH/bin is added to GITHUB_PATH so you can use commands such as nvcc directly in subsequent steps. Right now both windows-2019 and ubuntu-20.04 runners have been tested to work successfully.
Inputs
cuda
Optional The CUDA version to install. View src/link/windows-links.ts and src/link/linux-links.ts for available versions.
Default: '11.2.2'.
sub-packages
NOTE: On Linux this only works with the 'network' method view details
Optional
If set, only the specified CUDA subpackages will be installed.
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)
Default: '[]'.
method
Optional
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 withsub-packageson Windows).'network'downloads a smaller executable which only downloads necessary packages which you can define insub-packages.
Default: 'local'.
linux-local-args
Optional
(For 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)
See the Nvidia Docs for available options. Note that the --silent option is already always added by the action itself.
Default: '["--toolkit", "--samples"]'.
Outputs
cuda
The cuda version installed (same as cuda from input).
CUDA_PATH
The path where cuda is installed (same as CUDA_PATH in GITHUB_ENV).
Example usage
steps:
- uses: Jimver/[email protected]
id: cuda-toolkit
with:
cuda: '11.5.1'
- run: echo "Installed cuda version is: ${{steps.cuda-toolkit.outputs.cuda}}"
- run: echo "Cuda install location: ${{steps.cuda-toolkit.outputs.CUDA_PATH}}"
- run: nvcc -V