1.首先查看机器是否具有GPU
[javis@node5 rpm_resource]$ lspci | grep -i nvidia06:00.0 3D controller: NVIDIA Corporation GP100GL (rev a1)81:00.0 3D controller: NVIDIA Corporation GP100GL (rev a1)
- 1
- 2
- 3
- 4
2.在官网下载对应的驱动http://www.nvidia.cn/Download/index.aspx?lang=cn
然后跳转到下载页面,下载 nvidia-diag-driver-local-repo-rhel7-390.30-1.0-1.x86_64.rpm
(PS:注意下面红色的框内是安装步骤,安装后可以使用nvidia-smi命令)
3.安装步骤建议使用官方的
Once you accept the download please follow the steps listed below i) `rpm -i nvidia-diag-driver-local-repo-rhel7-390.30-1.0-1.x86_64.rpm' ii) `yum clean all`iii) `yum install cuda-drivers` iv) `reboot`
- 1
- 2
- 3
- 4
- 5
- 6
但这里安装的时候可能会遇到
Error: Package: 1:nvidia-kmod-390.30-2.el7.x86_64 (nvidia-diag-driver-local-390.30)Requires: dkms
- 1
下载安装dkms-2.2.0.3-31.1.noarch.rpm即可解决问题
sudo rpm -vih dkms-2.2.0.3-31.1.noarch.rpm
- 1
4.测试是否安装成功
[[email protected] cuda_example]$ nvidia-smiThu Mar 1 09:03:20 2018 +-----------------------------------------------------------------------------+| NVIDIA-SMI 387.26 Driver Version: 387.26 ||-------------------------------+----------------------+----------------------+| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC || Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. ||===============================+======================+======================|| 0 Tesla P100-PCIE... Off | 00000000:06:00.0 Off | 0 || N/A 28C P0 29W / 250W | 0MiB / 16276MiB | 0% Default |+-------------------------------+----------------------+----------------------+| 1 Tesla P100-PCIE... Off | 00000000:81:00.0 Off | 0 || N/A 26C P0 27W / 250W | 0MiB / 16276MiB | 0% Default |+-------------------------------+----------------------+----------------------++-----------------------------------------------------------------------------+| Processes: GPU Memory || GPU PID Type Process name Usage ||=============================================================================|| No running processes found |+-----------------------------------------------------------------------------+
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
PS:安装cuda的时候会动安装nvidia驱动,安装log信息
Installing the NVIDIA display driver...Installing the CUDA Toolkit in /usr/local/cuda-9.1 ...
- 1
- 2
- 3
- 4
- 5
cudann安装:
解压cudann.tgz包后后运行:sudo cp cuda/include/cudnn.h /usr/local/cuda/includesudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*