Skip to content

Enable CPU training #22

Description

@ronaldtse

Currently fails with no GPU:

 File "/usr/local/lib/python3.9/site-packages/torch/nn/modules/module.py", line 852, in to
return self._apply(convert)
File "/usr/local/lib/python3.9/site-packages/torch/nn/modules/module.py", line 530, in _apply
module._apply(fn)
File "/usr/local/lib/python3.9/site-packages/torch/nn/modules/module.py", line 552, in _apply
param_applied = fn(param)
File "/usr/local/lib/python3.9/site-packages/torch/nn/modules/module.py", line 850, in convert
return t.to(device, dtype if t.is_floating_point() or t.is_complex() else None, non_blocking)
File "/usr/local/lib/python3.9/site-packages/torch/cuda/__init__.py", line 166, in _lazy_init
raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled

Metadata

Metadata

Assignees

Labels

enhancementNew feature or request

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions