基于ubuntu安装cuda,cuDNN

2024-06-04 07:08
文章标签 ubuntu 安装 cuda cudnn

本文主要是介绍基于ubuntu安装cuda,cuDNN,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

1.cuda驱动安装

(1)下载驱动

NVIDIA 驱动程序下载, 根据自己的GPU型号旋转相应的驱动进行下载,下载后的驱动程序放入主目录下(/home/my_work),这是因为在后续的驱动纯命令行安装环境下,不支持中文输入,不容易切换目录。

 (2)卸载原有驱动

     两种方式,如果新装的的系统,就不必执行以下代码


#for case1: original driver installed by apt-get:
sudo apt-get remove --purge nvidia*#for case2: original driver installed by runfile:
sudo chmod +x *.run
sudo ./NVIDIA-Linux-x86_64-384.59.run --uninstall

(3)禁用nouveau驱动

sudo gedit /etc/modprobe.d/blacklist.conf

在最后一行添加:

blacklist nouveau

之后,执行命令:

sudo update-initramfs -u
电脑重启之后执行
lsmod | grep nouveau  #没有输出,即说明安装成功

(4)安装驱动

进入命令行界面:Ctrl-Alt+F6(我看有的是说ctrl-alt+F1,我测试是不可以的),之后输入用户名和密码登录即可(注意这里需要已经安装了gcc:sudo apt-get install build-essential),安装提示不断的接受即可安装成功。(尽量你的系统是英文版的,因为到了终端后,不能打汉字,要是你的系统是汉化的,那你就进不了目录里面)

sudo ./NVIDIA-Linux-x86_64-455.28.run
(或者sudo ./NVIDIA-Linux-x86_64-455.28.run --add-this-kernel)

安装完后,输入nvidia-smi,若输出GPU的状态信息就代表驱动安装成功,然后reboot重启

2.cuda安装

(1)根据gpu型号下载对应的驱动程序

CUDA Toolkit 11.0 Download | NVIDIA Developer

CUDA 工具包 - 免费工具和培训 | NVIDIA 开发者

根据需要决定是否安装10.2还是11.1版本的,右下角可以找到历史版本。

wget http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run
sudo sh cuda_11.0.2_450.51.05_linux.run

这里强烈建议选择deb(local)方式安装,因为后面如果使用tensorRT,他们提供的版本大多是是deb和tar,这里的runfile是什么鬼东西我也没搞明白)

【若在执行完上面第一条下载命令后出现了s段已转储错误,所以建议先执行以下步骤,没有出现可忽略此步骤:

gedit   .bashrc.

在文件末尾添加

ulimit -c unlimited
ulimit -s 819200

再另一个终端输入命令激活

source .bashrc

(2)安装cuda

  • 上面的第二条命令(sudo sh cuda_11.0.2_450.51.05_linux.run)里除了有一步输入accpet,以及下一步在很多X选择的时候,把第一个Nvidia显卡去掉(按回车去掉),其他都是默认,最后选install回车执行。

  • 安装完毕之后添加环境变量
  • gedit ~/.bashrc
    # 文本最后添加以下内容:
    export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
    export PATH=$PATH:/usr/local/cuda/bin
    export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
    # 保存退出,打开新终端激活
    source ~/.bashrc
    

    (3)测试cuda

  • cd /usr/local/cuda/samples/1_Utilities/deviceQuery
    sudo make
    ./deviceQuery

当输出Result = PASS字样如下则安装成功

CUDA安装成功后,即可进行GPU进行训练,后期安装的cuDNN是加速库,不安装也不影响gpu的使用

要想查看安装的cuda版本,可以用

nvcc -V   或者 cat /usr/local/cuda/version.txt , 正常情况下两个输出信息是一样的,第一个有事可能和第二个不一致,以第二个为准, 也就是你安装cuda的版本号 , nvidia-smi 界面显示的cuda版本不是你安装的cuda版本,

3. 安装cuDNN

(1)下载cuDNN

需要注册账号,下载cuDNN对应CUDA的版本

NVIDIA cuDNN | NVIDIA Developer

https://developer.nvidia.com/rdp/cudnn-download

(2)安装

可参考官方给出的安装指导, 对于linux系统,主要有两种方式,一种是tgz,一种是deb格式(要下载三个文件,runtime,dev,sample,并且依次安装),我尝试了用deb安装,安装成功后,

安装参考:Installation Guide :: NVIDIA Deep Learning cuDNN Documentation

(3)测试:

按照说明进行测试,编译一个sample的时候提示没有freeImage.h这个文件,。(这里也有可能会提示gcc版本过高的问题,若出现降低gcc,g++版本即可)


原因是确实相应的文件,解决方法是:

sudo apt-get install libfreeimage3 libfreeimage-dev

重新编译运行即可。

******

在安装cuda和cuDNN时尤其要注意版本之间的的兼容依赖关系,比如ubuntu版本,cuda,cuDNN,torch,gpu版本,如果哪里版本不对就可能出错,后面找问题很难,所以一定按官方给的版本一步步来。

参考:Ubuntu 20.04安装CUDA 11_绝版小哥的博客-CSDN博客_ubuntu安装cuda11

Ubuntu18.04下NVIDIA驱动+CUDA11.0安装 - it610.com

这篇关于基于ubuntu安装cuda,cuDNN的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/1029401

相关文章

如何在Ubuntu 24.04上部署Zabbix 7.0对服务器进行监控

《如何在Ubuntu24.04上部署Zabbix7.0对服务器进行监控》在Ubuntu24.04上部署Zabbix7.0监控阿里云ECS服务器,需配置MariaDB数据库、开放10050/1005... 目录软硬件信息部署步骤步骤 1:安装并配置mariadb步骤 2:安装Zabbix 7.0 Server

Ubuntu如何分配​​未使用的空间

《Ubuntu如何分配​​未使用的空间》Ubuntu磁盘空间不足,实际未分配空间8.2G因LVM卷组名称格式差异(双破折号误写)导致无法扩展,确认正确卷组名后,使用lvextend和resize2fs... 目录1:原因2:操作3:报错5:解决问题:确认卷组名称​6:再次操作7:验证扩展是否成功8:问题已解

Python中win32包的安装及常见用途介绍

《Python中win32包的安装及常见用途介绍》在Windows环境下,PythonWin32模块通常随Python安装包一起安装,:本文主要介绍Python中win32包的安装及常见用途的相关... 目录前言主要组件安装方法常见用途1. 操作Windows注册表2. 操作Windows服务3. 窗口操作

gitlab安装及邮箱配置和常用使用方式

《gitlab安装及邮箱配置和常用使用方式》:本文主要介绍gitlab安装及邮箱配置和常用使用方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录1.安装GitLab2.配置GitLab邮件服务3.GitLab的账号注册邮箱验证及其分组4.gitlab分支和标签的

MySQL MCP 服务器安装配置最佳实践

《MySQLMCP服务器安装配置最佳实践》本文介绍MySQLMCP服务器的安装配置方法,本文结合实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友参考下... 目录mysql MCP 服务器安装配置指南简介功能特点安装方法数据库配置使用MCP Inspector进行调试开发指

在Windows上使用qemu安装ubuntu24.04服务器的详细指南

《在Windows上使用qemu安装ubuntu24.04服务器的详细指南》本文介绍了在Windows上使用QEMU安装Ubuntu24.04的全流程:安装QEMU、准备ISO镜像、创建虚拟磁盘、配置... 目录1. 安装QEMU环境2. 准备Ubuntu 24.04镜像3. 启动QEMU安装Ubuntu4

Python UV安装、升级、卸载详细步骤记录

《PythonUV安装、升级、卸载详细步骤记录》:本文主要介绍PythonUV安装、升级、卸载的详细步骤,uv是Astral推出的下一代Python包与项目管理器,主打单一可执行文件、极致性能... 目录安装检查升级设置自动补全卸载UV 命令总结 官方文档详见:https://docs.astral.sh/

Nexus安装和启动的实现教程

《Nexus安装和启动的实现教程》:本文主要介绍Nexus安装和启动的实现教程,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录一、Nexus下载二、Nexus安装和启动三、关闭Nexus总结一、Nexus下载官方下载链接:DownloadWindows系统根

Java SWT库详解与安装指南(最新推荐)

《JavaSWT库详解与安装指南(最新推荐)》:本文主要介绍JavaSWT库详解与安装指南,在本章中,我们介绍了如何下载、安装SWTJAR包,并详述了在Eclipse以及命令行环境中配置Java... 目录1. Java SWT类库概述2. SWT与AWT和Swing的区别2.1 历史背景与设计理念2.1.

安装centos8设置基础软件仓库时出错的解决方案

《安装centos8设置基础软件仓库时出错的解决方案》:本文主要介绍安装centos8设置基础软件仓库时出错的解决方案,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐... 目录安装Centos8设置基础软件仓库时出错版本 8版本 8.2.200android4版本 javas