Kernel crashes

I successfully pulled a gpu enabled docker image with

nvidia-docker run --rm -p 10000:8888 -e JUPYTER_ENABLE_LAB=yes -v "$PWD":/workdir isbjornlabs/fastai-notebook-cuda9.2

This image is based on the fastai lesson plan and includes all the libraries.

I start a new notebook and try to follow along the steps in lesI changed the batch size with

son 1
This line crashes the kernel at about 6% into the first epoch.


I changed the batch size with
but it still crashes.

My gpu is a titanx pascal so I dont know why it should crash. I can run the same notebook fine in a kaggle kernel.

I discovered it was the docker image. Since switching to the one for paperspace-no problems.