I just setup a new machine of my own with core i5 and gtx 1070. I installed,NVIDIA drivers, CUDA and did fastai setup. However, when I am trying to run the example CATS and Dogs model, its taking forever to return any results after downloading the resnet model. How do I make sure the learner is using GPU and not CPU ?
Its working now. It turned out, just adding /usr/local/cuda-9.0 to PATH didn’t help. I now added a variable CUDA_HOME to .bashrc. Thanks for the pointers !!