欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Ubuntu16.04安装Cuda环境常用命令

程序员文章站 2022-05-09 22:01:09
...

基本环境及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/