davanstrien/grpo-completions
Updated
sudo apt -y install cuda-toolkit-12-8 nvidia-open
NVIDIA Graphics Device with CUDA capability sm_120 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_50 sm_60 sm_70 sm_75 sm_80 sm_86 sm_90.
pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu128