One cell in the notebook sets the CUDA_PATH environment variable to the default general CUDA installation directory:
import platform
if platform.system() == 'Linux':
%env CUDA_PATH=/usr/local/cuda
This will not necisarily be a CUDA 12 installation (as currently assumed), and also overrules any other user environment variables they may use to control this (e.g. via module files).
Additioanlly, this requires CUDA 12.x, it may be nice to
This should do some of the following:
- Use
/usr/local/cuda-12 if only supporting CUDA 12
- Use a user-specified version if one exists
- If CUDA_PATH is already set, use that
- If CUDA_HOME is specified, consider using that (the value set on windows, but some usage on linux too
- If neither set, check
PATH &/or run which nvcc and compute the path from that
- Ensure the path to be used exists
One cell in the notebook sets the
CUDA_PATHenvironment variable to the default general CUDA installation directory:This will not necisarily be a CUDA 12 installation (as currently assumed), and also overrules any other user environment variables they may use to control this (e.g. via module files).
Additioanlly, this requires CUDA 12.x, it may be nice to
This should do some of the following:
/usr/local/cuda-12if only supporting CUDA 12PATH&/or runwhich nvccand compute the path from that