安装CUDA8.0 cuDNN5.1

环境:GTX 1080, Ubuntu 16.04

准备

下载cuda-8.run file(保证MD5值和官网一致,为5639ffeb939ee58a81554d06bd084e15 cuda_8.0.27_linux.run)
下载Nvidia-367.18-driver.run file
删除旧的驱动 (sudo apt remove –purge nvidia*)
删除旧的nvidia.deb packages
删除cuda 文件夹(/usr/local/cuda*) (sudo apt remove –purge libcuda*)
reboot

安装驱动和cuda

进入文字界面 tty (ctrl-alt-1) ,
sudo service ligdm stop

run the Nvidia-367.18.run
run the cuda-8.run file.(注意:在某一步不要安装自带的361版本的驱动)

reboot
sudo service ligdm start

设置环境变量

sudo gedit /etc/profile
在最后加入以下两行,保存
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
然后使其生效, source /etc/profile

或者下面更潇洒(不过还没试验)
echo ‘export PATH=/usr/local/cuda/bin:$PATH’ >> ~/.bashrc
echo ‘export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH’ >> ~/.bashrc
source ~/.bashrc

校验

1)nvidia-smi
2)nvcc –version
3) 测试cuda的sample
cd /home/zhyj3038/NVIDIA_CUDA-8.0_Samples
sudo make
然后
cd /NVIDIA_CUDA-8.0_Samples
1)./bin/x86_64/linux/release/deviceQuery
2)./5_Simulations/nbody/nbody

cuDNN5.1

tar -zxvf cudnn-8.0-linux-x64-v5.1.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

你可能感兴趣的:(mxnet安装)