基本环境及cuda(包括NVIDIA驱动以及cudnn)
1、制作U启动,安装好系统
2、sudo apt update
3、sudo apt upgrade
#(此操作更新列表里可更新的软件)
4、sudo apt-get install build-essential
#(安装一些基础的环境)
方案一:(该方法会把nVidia驱动同时装上)
5、下载cuda8.0的DEB local安装包
6、sudo dpkg -i cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.deb
7、sudo apt update
8、sudo apt install cuda
9、reboot
方案二:(该方法会让你选择是否需要安装显卡驱动)
5、下载cuda8.0的runfile local(即.run)安装包
6、sudo sh cuda_9.0.176_384.81_linux.run
7、按照提示和自己的需求安装(工具、驱动等)
8、9、reboot
10、往~/.bashrc中添加cuda环境变量
export CUDA_HOME=/usr/local/cuda-9.0
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
11、查看环境中cuda版本
nvcc -V
12、验证cuda安装正确性:
cd到cuda测试例子目录(eg:/usr/local/cuda-9.0/samples/1_Utilities/deviceQuery)
sudo make
./deviceQuery
看到输出中有:
Result = PASS
恭喜你,cuda安装成功
13、解压cudnn包
14、放在指定位置例如:(或/usr/lib/x86_64-linux-gnu)
sudo cp include/cudnn.h /usr/local/cuda-9.0/include/
sudo cp lib64/* /usr/local/cuda-9.0/lib64/