doremi
doremi copied to clipboard
Cuda version problem
I'm trying to run this script on aws sagemaker
cd doremi && bash scripts/setup_flash.sh
it throws an error:
The detected CUDA version (12.1) mismatches the version that was used to compile
PyTorch (11.7). Please make sure to use the same CUDA versions.