Ubuntu Core 14.04 + CUDA 8.0 + cuDNN v5 + Theano.
163
I tried to run the following theano image with gtx 1080:
https://hub.docker.com/r/kaixhin/cuda-theano/
Unfortunately, faced two errors:
ERROR (theano.sandbox.cuda): ERROR: Not using GPU. Initialisation of device gpu failed: initCnmem: cnmemInit call failed! Reason=CNMEM_STATUS_OUT_OF_MEMORY. numdev=1
nvcc fatal : Value 'sm_61' is not defined for option 'gpu-architecture'
Error 1 can be solved by specify CNMEM=0.8 in the .theanorc Error 2 can be solved by cuda-8.0 install based on:
Now, it works like a charm.
nvidia-docker run -it fananymi/theano:cuda-8.0
Content type
Image
Digest
Size
2 GB
Last updated
about 10 years ago
docker pull fananymi/theano:cuda-8.0