cuda周辺環境を整える

1.既存ドライバ等の確認

dpkg -l | grep nvidia
dpkg -l | grep cuda

2.cuda,nvidiaのアンインストー

sudo apt-get --purge remove nvidia-*
sudo apt-get --purge remove cuda-*

 

3.ドライバーのインストー

sudo apt-get install nvidia-3**

4.CUDA8.0のインストー

https://developer.nvidia.com/cuda-downloads より


sudo dpkg -i cuda-repo-ubuntu1604_8.0.*_amd64.deb
sudo apt update
sudo apt install cuda

5.setting path

echo -e "\n## CUDA and cuDNN paths"  >> ~/.bashrc
echo 'export PATH=/usr/local/cuda-8.0/bin:${PATH}' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:${LD_LIBRARY_PATH}' >> ~/.bashrc
source ~/.bashrc # CUDAのパスが書き込まれた~/.bashrcを読み込む。

6.sudo reboot

sudo reboot

7.nvcc -V

nvcc -V

8.cudnn

cuDNN 5.1 for CUDA 8.0をダウンロードする(アクセスするにはメンバー登録が必要)。

https://developer.nvidia.com/rdp/cudnn-download
よりダウンロード
tar xzvf cudnn-8.0-linux-x64-v5.1.tgz 
sudo cp -a cuda/lib64/* /usr/local/cuda-8.0/lib64/
sudo cp -a cuda/include/* /usr/local/cuda-8.0/include/
sudo ldconfig

環境

ubuntu16.04LTS
GTX-1080