Closed kpaonaut closed 6 years ago
This is a problem resulted from conflicted cuda version or some problems of cuda. I would suggest anyone has this problem reinstall cuda 8.0(9.0 will not work!) and also reboot their machine to make sure NVIDIA works fine.
Any tutorial to switch to CPU-only caffe?caffe.set_mode_cpu()
seems not enough. I am studying another problem anyway.
I was able to run make_lmdb.sh with cuda8, however, this shell gives me this error when running with cuda 9:
I'm not sure if this is the problem of cuda version, or something else. This error alone does not prevent the shell from running, but it makes train.sh aborted:
Any thoughts? Thank you!