Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

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

like image 579
Joshua Owoyemi Avatar asked Jun 30 '16 13:06

Joshua Owoyemi


2 Answers

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.

like image 195
Tal Ben-Nun Avatar answered Nov 19 '22 01:11

Tal Ben-Nun


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

like image 33
Joshua Owoyemi Avatar answered Nov 19 '22 00:11

Joshua Owoyemi