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
As Robert Crovella said, SM 6.1 (sm_61) is only supported in CUDA 8.0 and above, and thus you should download CUDA 8.0 Release Candidate from https://developer.nvidia.com/cuda-toolkit
Ubuntu 14.04 is supported, and the instructions on the website on how to setup should be straightforward (copy and paste lines to the console).
I would also recommend downloading CUDA 8.0 when it comes out, since the RC is not the final version.
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
If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!
Donate Us With