Sign inSign up

fananymi/theano

By fananymi

•Updated about 10 years ago

Ubuntu Core 14.04 + CUDA 8.0 + cuDNN v5 + Theano.

Image
0

163

fananymi/theano repository overview

⁠Background

I tried to run the following theano image with gtx 1080:

https://hub.docker.com/r/kaixhin/cuda-theano/⁠

Unfortunately, faced two errors:

  1. 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

  2. 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:

http://stackoverflow.com/questions/38125310/nvcc-fatal-value-sm-61-is-not-defined-for-option-gpu-architecture-error-wi⁠

Now, it works like a charm.

⁠Usage

nvidia-docker run -it fananymi/theano:cuda-8.0

Tag summary

Content type

Image

Digest

Size

2 GB

Last updated

about 10 years ago

docker pull fananymi/theano:cuda-8.0