This repository contains some sample code that's helpful for getting started with using CUDA on Modal.
The subdirectories include:
-
cuda
: A demonstration of how to compile and run custom CUDA code on Modal. -
cupy
: A simple script showing the use of CuPy on Modal. -
jupyter_cuda_mode
: Two example notebooks from the CUDA MODE class plus a script for running them on Modal. These notebooks show workflows for using CUDA from Python. -
vscode_on_modal
: A simple script for launching a VS Code instance on Modal.
For usage information, see the script files in each folder.
You can find a lot more examples of how to use Modal in the examples repository.