Ubuntu18中NVIDIA,cuda,cudnn,pytorch安装

2023-11-13

注意:nvidia驱动和cuda,cudnn,pytroch,python的对应关系

参考内容:

https://blog.csdn.net/tjuyanming/article/details/80862290

https://blog.csdn.net/weixin_43162240/article/details/102957660

安装流程:安装cuda(包括nvidia驱动)+ cudnn + python安装

注意:nvidia驱动可以在安装cuda时一起安装

1、安装NVIDIA驱动(该驱动也可以在cuda安装时一起安装)

Ubuntu 18.04 NVIDIA驱动安装总结_ObsessionLife的博客-CSDN博客_ubuntu18.04 安装nvidia显卡驱动

最近遇到了在Ubuntu 18.04上安装NVIDIA的情况,看到一篇英文教程讲解的很好,简单翻译一下拿来收藏。

在Ubuntu 18.04上安装NVIDIA有三种方法:

使用标准Ubuntu仓库进行自动化安装

使用PPA仓库进行自动化安装

使用官方的NVIDIA驱动进行手动安装

上述三种方法均可用,我个人更习惯于使用手动安装。

注意:

在安装之前首先就是要禁用Nouveau的驱动,禁用该驱动的方法参照这篇博客。

ubuntu禁止内核自动更新 - 学渣Mia - 博客园

ubuntu禁止自动更新内核_maohule的博客-CSDN博客_ubuntu禁止自动更新内核 maohule

1.查看系统安装的内核有哪些

  dpkg --get-selections | grep linux-image

2.查看系统正在使用的内核

 

 uname -a

禁止更新内核:

sudo apt-mark hold linux-image-4.13.0-36-generic
sudo apt-mark hold linux-image-extra-4.13.0-36-generic
sudo apt-mark hold linux-headers-4.13.0-36-generic

如果需要恢复原来的设定的话即允许更新内核,执行如下命令即可:

sudo apt-mark unhold linux-image-4.13.0-36-generic
sudo apt-mark unhold linux-image-extra-4.13.0-36-generic

————————————————

上一步的改动只是在安装的时候临时禁用。如果没有永久禁用该驱动,可能会出现安装完毕NIVIDA显卡后无法进入Ubuntu的情况(在登录界面,输入密码也无法登录)。

所以,在安装后Ubuntu成功后需要在grub的配置文件里面更改:

在文本中搜索quiet splash 然后添加acpi_osi=linux nomodeset,保存文本即可。

1. 使用标准Ubuntu 仓库进行自动化安装

这种方法几乎是所有的示例中最简单的方法,也是该教程最为推荐的方法。首先,检测你的NVIDIA显卡型号和推荐的驱动程序的模型。在命令行中输入如下命令:

== cpu-microcode.py ==

1. 使用标准Ubuntu 仓库进行自动化安装

这种方法几乎是所有的示例中最简单的方法,也是该教程最为推荐的方法。首先,检测你的NVIDIA显卡型号和推荐的驱动程序的模型。在命令行中输入如下命令:

$ ubuntu-drivers devices
== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001180sv00001458sd0000353Cbc03sc00i00
vendor   : NVIDIA Corporation
model    : GK104 [GeForce GTX 680]
driver   : nvidia-304 - distro non-free
driver   : nvidia-340 - distro non-free
driver   : nvidia-384 - distro non-free recommended
driver   : xserver-xorg-video-nouveau - distro free builtin
 
== cpu-microcode.py ==
driver   : intel-microcode - distro free

从输出结果可以看到,目前系统已连接Nvidia GeFrand GTX 680显卡,建议安装驱动程序是 nvidia-384版本的驱动。如果您同意该建议,请再次使用Ubuntu驱动程序命令来安装所有推荐的驱动程序。

输入以下命令:

sudo ubuntu-drivers autoinstall

一旦安装结束,重新启动系统(不重启可能会出现nvidia不匹配等错误),

sudo reboot

你就完成了。

输入

nvidia-smi

出现一些现实,说明驱动安装正确

sudo apt install nvidia-driver-418

#去除之前的nvidia显卡驱动

sudo apt-get autoremove --purge nvidia-*

sudo apt-get remove --purge 'nvidia\.*'

sudo apt autoremove

一旦安装结束,重新启动系统(不重启可能会出现nvidia不匹配等错误),

CUDA Toolkit and Minimum Compatible Driver Versions

CUDA Toolkit

Linux x86_64 Driver Version

Windows x86_64 Driver Version

CUDA 11.2.2 Update 2

>=460.32.03

>=461.33

CUDA 11.2.1 Update 1

>=460.32.03

>=461.09

CUDA 11.2.0 GA

>=460.27.03

>=460.82

CUDA 11.1.1 Update 1

>=455.32

>=456.81

CUDA 11.1 GA

>=455.23

>=456.38

CUDA 11.0.3 Update 1

>= 450.51.06

>= 451.82

CUDA 11.0.2 GA

>= 450.51.05

>= 451.48

CUDA 11.0.1 RC

>= 450.36.06

>= 451.22

CUDA 10.2.89

>= 440.33

>= 441.22

CUDA 10.1 (10.1.105 general release, and updates)

>= 418.39

>= 418.96

CUDA 10.0.130

>= 410.48

>= 411.31

CUDA 9.2 (9.2.148 Update 1)

>= 396.37

>= 398.26

CUDA 9.2 (9.2.88)

>= 396.26

>= 397.44

CUDA 9.1 (9.1.85)

>= 390.46

>= 391.29

CUDA 9.0 (9.0.76)

>= 384.81

>= 385.54

CUDA 8.0 (8.0.61 GA2)

>= 375.26

>= 376.51

CUDA 8.0 (8.0.44)

>= 367.48

>= 369.30

CUDA 7.5 (7.5.16)

>= 352.31

>= 353.66

CUDA 7.0 (7.0.28)

>= 346.46

>= 347.62

2、安装CUDA11.1

ubuntu安装cuda11.2_AI界扛把子的博客-CSDN博客

linux安装pytorch(包括cuda与cudnn)_Bryce__LY的博客-CSDN博客

CUDA Toolkit 11.8 Downloads | NVIDIA Developer

  1. 官网地址

官网下载cuda,特别注意:要选择run,不要选择deb。

wget http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run
chmod +x cuda_10.1.105_418.39_linux.run

sudo ./cuda_10.1.105_418.39_linux.run

这里是下载的cuda 11.1版本

  1. 禁用X服务(黑屏)
  1. 给文件赋执行权限并安装

后续选择accepted,

除了驱动不选择(因为第一步已经安装了nvidia驱动了,如果前面没有安装nvidia,则该选项也需要),其他都选,具体界面如下

打开配置文件,在终端输入

vim ~/.bashrc

在末尾添加写入(这个相当于选择cuda-11.1版本)

export PATH=/usr/local/cuda-11.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64:$LD_LIBRARY_PATH

更新一下

输入nvcc -V,显示cuda版本,检测是否成功

查看版本信息

whereis cuda
whereis cudnn

cuda11.1及以下的卸载:

cuda10.0及以下的卸载:

cd /usr/local/cuda-xx.x/bin/ sudo ./uninstall_cuda_xx.x.pl sudo rm -rf /usr/local/cuda-xx.x

sudo apt-get --purge remove "*cublas*" "cuda*"

3、安装cudnn

  • 下载对应cuda 11.1 的cudnn版本

官网下载,找到与cuda版本配对的cudnn,查看官网配对,以及更全的版本配对,我的cuda为11.1,选择对应的cudnn版本

cuDNN Archive | NVIDIA Developer

  • 将下载的添加为压缩包并解压 (不同版本,则将cunn-11.1中11.1修改其他版本的数字即可)
  • 查看安装的cudnn版本

  • cudnn的卸载

如果有之前的残留版本最好卸载干净,同时路径中的cuda字样,要根据现实来定,cuda有可能需要替换为cuda-9.0或者其他

sudo rm -rf /usr/local/cuda-11.1/include/cudnn.h

sudo rm -rf /usr/local/cuda-11.1/lib64/libcudnn*

四、安装pytroch

根据网址Start Locally | PyTorch选择配置

五、验证

以上安装完成后,打开python,执行以下脚本。

若输出true,则表示cuda可以使用了。然后开始愉快的torch之旅吧。

方法1:查看所有python的路径,不止一个

方法2:查看当前使用的python路径

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

Ubuntu18中NVIDIA,cuda,cudnn,pytorch安装 的相关文章

随机推荐

  • Keil最舒服的配色方案

    每次重装keil 都会重新配置下颜色 因为默认的颜色 看久了实在不太舒服 每次调到一个舒服的颜色 都要花不少时间 为了后面节省时间 现记录一下 如下就是我觉得还不错的配色方案 如果你觉得也还可以 可以接着往下看 一 进入配置 二 光标所在行
  • 数据预处理与可视化

    https blog csdn net u013719780 article category 6285514 https blog csdn net u013719780 article category 6457723
  • 公司新招了个字节拿36K的人,让我见识到了什么才是测试扛把子......

    5年测试 应该是能达到资深测试的水准 即不仅能熟练地开发业务 而且还能熟悉项目开发 测试 调试和发布的流程 而且还应该能全面掌握数据库等方面的技能 如果技能再高些的话 甚至熟悉分布式组件等高级技能 或者说 做个项目小组长 管个3 4号人 应
  • SpringBoot-yml配置文件的使用与优势

    目录 一 什么是yml配置文件 二 如何在SpringBoot中使用yml配置文件 三 如何在SpringBoot中注入yml配置文件的属性 Value注解 ConfigurationProperties注解 四 yml配置文件与prope
  • c++ 成员属性设置为私有

    成员属性设置为私有 1 可以自己控制读写权限 2 对于写可以检测数据的有效性 include
  • 免费版Typora设置vue主题

    一 下载Typora与主题vue安装包 最新版版本的Typora网上开始收取费用 现在低版本的Typora仍然还是免费的 而且并不影响正常使用 下载链接在下方 并提供了vue主题文件 链接 百度网盘 请输入提取码 提取码 tdp4 二 添加
  • 【OpenCV图像处理】1.15自定义线性滤波

    自定义线性滤波 线性滤波就是卷积 实际上的作用就是降噪 卷积概念 卷积是图像处理中一个操作 是kernel在图像的每个像素上的操作 Kernel本质上一个固定大小的矩阵数组 其中心点称为锚点 anchor point 卷积如何工作 把ker
  • ggplot2读书笔记11:第八章 主题设置、存储导出

    Themes 8 1 Introduction 第八章讲的是ggplot2的主题设置 通过它你可以对数据之外的图形外观进行控制 第一版的中文版的把这一章的章节名翻译为 精雕细琢 控制主题设置主要有以下四个方面 主题元素 指的是非数据元素 p
  • 【吴恩达机器学习 - 5】揭秘学习曲线:偏差和方差与代价函数之间的相互关系(课后练习第五题-ex5)

    目录 1 回顾线性回归的正则化 1 1 代价函数 1 2 梯度 2 偏差与方差 2 1 训练集数量对 代价 的影响 3 多项式回归 3 1 多项式特征的生成 3 2 lambda对 代价 的影响 照旧 本练习的相关资料链接将会扔到评论区 大
  • cocos2d-x学习之路(14)--屏幕适配

    屏幕适配是所有手机应用都需要解决的问题 在cocos2d x 2 0之后的版本提供了相应的解决方法 设置屏幕分辨率 cocos2d x 2 1版本设置分辨率在main cpp中 通过改变FrameSize模拟相应的分辨率 CCEGLView
  • 【GoCN译文视点】何时使用 Rust, 何时使用 Go

    原文地址 https medium com codex when to use rust and when to use go 590bcbb49bec 原文作者 Joel 本文永久链接 Github 译者 tt 直截了当的说 Go 和 R
  • Flutter全部图标之快速找到合适的Icon图标

    1 由于Flutter中的图标太多 并且无法直观地进行预览 因为把所有的图标截图下来展示 以便可以快速找到你想要的图标
  • 2023.6.26

  • jdk安装

    开发十年 就只剩下这套Java开发体系了 gt gt gt 1 安装jdk 1 1 jdk下载地址 http www oracle com technetwork java javase downloads jdk8 downloads 2
  • 关于服务连接器(Servlet)你了解多少?

    文章目录 Servlet 1 简介 2 快速入门 3 执行流程 4 生命周期 5 方法介绍 6 体系结构 7 urlPattern配置 8 XML配置 最后说一句 Servlet 1 简介 Servlet是JavaWeb最为核心的内容 它是
  • [Ctf show web] 萌新计划12

    preg match 没有匹配到表达式时输出 c 利用baes64加密绕过 c a base64 decode c3lzdGVt b base64 decode dGFjIGNvbmZpZy5waHA a b
  • 1127: 矩阵乘积java

    import java util Scanner public class Main public static void main String args Scanner input new Scanner System in int m
  • springboot+redistemplate 集群配置

    redisConfig 配置类 package com zhouzhou springredistemplatedemo controller import io lettuce core cluster ClusterClientOpti
  • 前言

    es7 1中文指南 前言 数据存储 信息输出 搜索 分析 more 前言 Elasticsearch是Elastic Stack核心的分布式搜索和分析引擎 Logstash和Beats有助于收集 聚合和丰富您的数据并将其存储在Elastic
  • Ubuntu18中NVIDIA,cuda,cudnn,pytorch安装

    注意 nvidia驱动和cuda cudnn pytroch python的对应关系 参考内容 https blog csdn net tjuyanming article details 80862290 https blog csdn