Fixed it! I had to get cuda 10.0 (specifically). Used deb packages. Follow the instructions here, but with cuda 10.0:
https://www.tensorflow.org/install/gpu
Batches run faster (I think), but there is a longer delay between batches, presumably while the job is parallelized???
No comments:
Post a Comment