When I trained a model, I got the below error. I'm using a Mac book pro. Thanks.
File "/usr/local/lib/python2.7/site-packages/torch/cuda/__init__.py", line 277, in __new__
_lazy_init()
File "/usr/local/lib/python2.7/site-packages/torch/cuda/__init__.py", line 89, in _lazy_init
_check_driver()
File "/usr/local/lib/python2.7/site-packages/torch/cuda/__init__.py", line 56, in _check_driver
raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled
I added --gpu_ids -1 and this issue was resolved. Thanks.
I added
--gpu_ids -1and this issue was resolved. Thanks.
where you added that
have the same issue, where you added that ?
You need to add it in the command line.
Most helpful comment
where you added that