Ubuntu+CUDA9.0 安装

在terminal里面输入命令nvidia-smi,看到如下的输出,就是正确了。

 

二、CUDA9.0下载。地址:https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=deblocal

 

AB安装方式二选一

A、run安装方式:

下载完CUDA 9.0之后执行如下语句,运行.run文件

    sudo sh cuda_9.0.176_384.81_linux.run

单击回车,一路往下运行,直到提示“是否为NVIDIA安装驱动nvidia-384?”,选择否,因为已经安装好驱动程序了,其他的全都是默认,不过要记住安装位置,默认是安装在/usr/local/cuda文件夹下。

B、deb安装

打开终端,依次输入下面的命令:

sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda

三、配置环境变量

配置环境变量,运行如下命令打开profile文件

    sudo gedit /etc/profile

  打开文件后在文件末尾添加路径,也就是安装目录,命令如下:

    export PATH=/usr/local/cuda-9.0/bin:$PATH

    export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH  

  保存,然后重启电脑

    sudo reboot

四、检查是否安装成功

a、 验证驱动版本
$ cat /proc/driver/nvidia/version
结果显示类似
NVRM version: NVIDIA UNIX x86_64 Kernel Module 384.81 Sat Sep 2 02:43:11 PDT 2017
GCC version: gcc version 5.4.0 20160609 (Ubuntu 5.4.0-6ubuntu1~16.04.5)

b、 验证CUDA Toolkit
$ nvcc -V 会输出CUDA的版本信息

如果是这样的:
The program 'nvcc' is currently not installed. You can install it by typing:
sudo apt-get install nvidia-cuda-toolkit

可能是环境配置没有成功,重新配置环境。

五、测试CUDA的Samples例子

  cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery

   sudo make

   ./deviceQuery

  如果显示的是关于GPU的信息,则说明安装成功了。

posted @ 2021-10-21 13:04  马语者  阅读(235)  评论(0编辑  收藏  举报