Search code examples
pythoncudagputheanonvcc

nvcc fatal : Value 'sm_61' is not defined for option 'gpu-architecture' error with theano

I was setting up python and theano for use with gpu on; ubuntu 14.04, GeForce GTX 1080 already installed NVIDIA driver (367.27) and CUDA toolkit (7.5) successfully for the system, but on testing with theano gpu implementation I get the above error (for example; when importing theano with gpu enabled) I have tried to look for possible solutions but didn't succeed. I'm a little new to ubuntu and gpu programming, so I would appreciate any insight into how I can solve this problem. Thanks

Solution

  • I was able to find a solution to this problem (since I still want to use CUDA 7.5) by including the following line in the .theanorc file

    flags = -arch=sm_52

    no more nvcc fatal error