nanoGPT icon indicating copy to clipboard operation
nanoGPT copied to clipboard

no cuda training does not work.

Open BurkenDev opened this issue 3 months ago • 1 comments

This needs fixing since that package in debian 12 isn't compiled with cuda, you should have this be able to train without cuda as well. Up to us to choose whether we want cpu or gpu training.

Traceback (most recent call last): File "/root/nanoGPT/train.py", line 193, in model.to(device) File "/usr/lib/python3/dist-packages/torch/nn/modules/module.py", line 989, in to return self._apply(convert) ^^^^^^^^^^^^^^^^^^^^ File "/usr/lib/python3/dist-packages/torch/nn/modules/module.py", line 641, in _apply module._apply(fn) File "/usr/lib/python3/dist-packages/torch/nn/modules/module.py", line 641, in _apply module._apply(fn) File "/usr/lib/python3/dist-packages/torch/nn/modules/module.py", line 664, in _apply param_applied = fn(param) ^^^^^^^^^ File "/usr/lib/python3/dist-packages/torch/nn/modules/module.py", line 987, in convert return t.to(device, dtype if t.is_floating_point() or t.is_complex() else None, non_blocking) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/usr/lib/python3/dist-packages/torch/cuda/init.py", line 221, in _lazy_init raise AssertionError("Torch not compiled with CUDA enabled") AssertionError: Torch not compiled with CUDA enabled

BurkenDev avatar Mar 20 '24 20:03 BurkenDev