ubuntu14.04へのcudaを再インストール

普段からcudaを使っていないと、たまにcudaを実行してみると上手くいかないことが多い。まずは、以下のソースを実行して、GPUにメモリを確保できているか確かめる。

include <stdio.h>

int main(void){
    int *dev_c;
    cudaError_t err;
   
    //GPU上にメモリが確保できているか確認する。
    err=cudaMalloc((void **)&dev_c,sizeof(int));

    if(err!=cudaSuccess){
        printf("error\n");
    }

    return 0;
}

これで、errorが出力されるとドライバーやコンパイラのアップデートが必要となると思う。
まずは、ドライバーから...。と、その前にGPUの型番の確認

sudo lshw -short -class display

www.nvidia.co.jp

次に、コンパイラーdeveloper.nvidia.com

ドライバーをインストールするためにX serverを切る必要があります。

sudo service lightdm stop

早速、ドライバーをインストール

chmod +x NVIDIA-Linux-x86_64-346.59.run
sudo ./NVIDIA-Linux-x86_64-346.59.run

次に、コンパイラーをインストール

chmod +x NVIDIA-Linux-x86_64-340.24.run
sudo ./NVIDIA-Linux-x86_64-340.24.run

.bashrcに以下を書きくわる

export PATH=/usr/local/cuda-7.0/bin:$PATH                                                                                                                                                      
export LD_LIBRARY_PATH=/usr/local/cuda-7.0/lib64:$LD_LIBRARY_PATH

問題なければ、

sudo service lightdm start

Xをスタートしていないと、以下のようなエラーが出るので注意

modprobe: ERROR: could not insert 'nvidia_331_uvm': Invalid argument