Centos 7 使用CUDA进行GPU加速运算

谁借莪１个温暖的怀抱￠ 2023-02-23 12:30 11阅读 0赞

### 1.需要一个Nvidia的显卡（暂时没发现可以使用虚拟机来直接操作显卡进行计算） ###

# #

### 2.需要安装NVIDIA的驱动 ###

一.找到你对应显卡的驱动，下载地址如下

[https://www.nvidia.co.uk/Download/index.aspx?lang=en-uk][https_www.nvidia.co.uk_Download_index.aspx_lang_en-uk]

最后下载到这样的文件

![20200707200517708.png][]

二.给.run文件添加执行权限

cd 下载文件夹路径（进入文件夹）

使用chmod 777 NVIDIA-Linux-x86\_64-440.100.run赋予执行权限，或者右击 -->properties -->permission -->execute打勾

三. 安装依赖项

需要有gcc，对应系统版本的kernel-devel， 对应系统版本的kernel-headers

`自己系统的版本可使用bash命令uname` `-r查询`

`rpm包下载地址`

[http://rpm.pbone.net/][http_rpm.pbone.net]

`下载好后进入文件目录`

`执行安装命令，例sudo` `yum install`kernel-devel-3.10.0-1062.el7.x86\_64.rpm``

四. 禁用nouveau

注：一般Linux安装的时候默认会安装nouveau驱动。 这个驱动会与Nvidia官方的驱动冲突，

`进入root模式 su roor`

`创建文件`/etc/modprobe.d/blacklist-nouveau.conf

在文件中添加如下内容

blacklist nouveau

options nouveau modeset=0

保存后执行

mv vmlinuz-$(uname -r)  vmlinuz-$(uname -r) .bak.nouveau 做好备份

dracut --v   vmlinuz-$(uname -r)  $(uname -r)  生成新文件

重启reboot，完成后使用bash命令：`lsmod | grep nouveau，当成功禁用后，不会有任何输出。`

`五.安装驱动`

`su root 进入root身份`

systemctl set-default multi-user.target  设置默认命令行启动，这样作可以避免万一驱动有问题时无法进入系统的尴尬

systemctl isolate multi-user.target  进入命令行模式

cd 显卡驱动目录

执行安装程序  ./NVIDA-Linux-x86\_64-440.100.run（注：尽量下载最新的驱动，因为驱动向前兼容）

安装时给出选项都点no（尤其不要改变x - config），不然有可能桌面启动不了

至此驱动安装全部完成

reboot 重启后登陆

systemctl isolate graphical.target启动图形模式

成功后systemctl set-default graphical.target 设置启动后直接进入桌面

注： 卸载驱动  ./NVIDIA-Linux-x86\_64-440.100.run  --uninstall

### 3.安装CUDA和cuDNN ###

1.打开ananconda navigtor，进入environment，安装如下两个包

![20200707211214690.png][]

大工告成，在对应的anaconda环境中可以愉快地加速计算神经网络了（linux不要随便升级内核，cuda会直接不能用的）

如果出现运行时cuda使用失败，需要查看以下cuda官网文档

[https://developer.nvidia.com/cuda-toolkit-archive][https_developer.nvidia.com_cuda-toolkit-archive]

一般情况下应该根据自己的系统先选择好cuda版本，然后选定显卡驱动（版本不能太低），然后选定pytorch或其他框架版本。

[https_www.nvidia.co.uk_Download_index.aspx_lang_en-uk]: https://www.nvidia.co.uk/Download/index.aspx?lang=en-uk
[20200707200517708.png]: https://img-blog.csdnimg.cn/20200707200517708.png
[http_rpm.pbone.net]: http://rpm.pbone.net/
[20200707211214690.png]: https://img-blog.csdnimg.cn/20200707211214690.png
[https_developer.nvidia.com_cuda-toolkit-archive]: https://developer.nvidia.com/cuda-toolkit-archive