当前位置: 首页 > news >正文

ubuntu, nvidia driver, cuda, cudnn, pytorch-gpu版本安装

文章目录

  • 1.常用指令
      • 1.1查看cpu是intel还是amd:
      • 1.2.查看ubuntu版本
      • 1.3.查看架构
      • 1.4.查看已安装的nvidia驱动
      • 1.5.进入tty模式
  • 2.安装ubuntu22.04 和 nvidia 驱动
  • 3.ubuntu 安装 anaconda
  • 4.安装pytorch gpu版本
  • 5.安装完整版cuda 和 cudnn
  • 6.nvidia-driver, cuda-toolkit, cudnn

1.常用指令

1.1查看cpu是intel还是amd:

cat /proc/cpuinfo
或者lscpu

1.2.查看ubuntu版本

lsb_release -a

1.3.查看架构

arch
或者
uname -a

1.4.查看已安装的nvidia驱动

nvidia-detector

1.5.进入tty模式

Ctrl + Alt + F3进入tty模式
Ctrl + Alt + Delete 退出tty模式

2.安装ubuntu22.04 和 nvidia 驱动

https://blog.csdn.net/takedachia/article/details/130354110

3.ubuntu 安装 anaconda

https://developer.aliyun.com/article/988482
export PATH=“~/anaconda3/bin”:$PATH 导入变量 source ~/.bashrc使生效

conda create -n pytorch python=3.9

4.安装pytorch gpu版本

记住此时安装的cuda是不完整的,但是对于pytorch来说够用了。
https://pytorch.org/get-started/previous-versions/
可以看到:
conda install pytorch1.13.1 torchvision0.14.1 torchaudio==0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia
这样安装以后gpu版本的pytorch就可以使用了。
在这里插入图片描述

但是此时 /usr/local/ 下面并没有cuda.

关于 nvidia-smi 和 nvcc --version 以及 print(torch.version.cuda) 不一致等cuda版本相关问题查看:
https://www.jb51.net/article/278075.htm
https://www.jianshu.com/p/eb5335708f2a
https://www.cnblogs.com/yhjoker/p/10972795.html

5.安装完整版cuda 和 cudnn

参考:
https://blog.csdn.net/takedachia/article/details/130375718
介绍的很详细。
上面的教程是通过https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html Package Manager Installation的方法安装cudnn的

安装后的 头文件和so文件 在
在这里插入图片描述

也可以利用tar file installation方法:
不需要安装,只需要解压和权限设置即可
在这里插入图片描述

6.nvidia-driver, cuda-toolkit, cudnn

除了以上的 通过下runfile来安装,也可以通过包安装
比如安装nvidia driver:
ubuntu-drivers devices 寻找可以安装的驱动

sudo ubuntu-drivers autoinstall 安装推荐
或者
sudo apt-get install cuda-drivers-418 安装特定版本的驱动。
nvidia-smi确认驱动安装正确

安装cuda:
conda search -c conda-forge cudatoolkit # 列出所有可以安装的版本 conda install -c conda-forge cudatoolkit=xx.x.x

nvcc-V 确定安装成功

cudnn:也有多种安装方法
https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html

http://www.lryc.cn/news/223692.html

相关文章:

  • 探索环保葡萄酒之生物动力
  • 【线上问题】服务器关机导致docker启动的mysql数据库消失了
  • Win10 180天后怎么才能继续体验,自动保持续期,无需手动JH
  • RHCE8 资料整理(五)
  • CPU 飙高系统反应慢怎么排查
  • 深度学习之基于YoloV5-Deepsort人物识别与追踪系统
  • Spring Boot中配置多个数据源
  • C++学习笔记---命名空间namespace
  • 缓存-Spring Cache 缓存抽象
  • Java修仙传之神奇的ES2(巧妙的查询及结果处理篇)
  • 架构设计的课程资料
  • 数据结构与算法C语言版学习笔记(5)-串,匹配算法、KMP算法
  • 新版HI3559AV100开发注意事项
  • Django(一、简介,安装与使用)
  • 【Linux C IO多路复用】多用户聊天系统
  • JSON——数组语法
  • 运营商大数据精准获客:我们提供精准客源渠道的最大资源体?
  • 表象变换与矩阵元
  • vue乾坤微前端项目
  • 大语言模型比武
  • 王道数据结构第五章二叉树的遍历第13题
  • 微服务的发展历程的详细说明及每个阶段主流的架构和组件
  • 2023年眼镜行业分析(京东眼镜销量数据分析):市场规模同比增长26%,消费需求持续释放
  • 基础课26——业务流程分析方法论
  • 【数字图像处理-TUST】实验二-图像噪声生成与滤波降噪
  • bilibili快速升满级(使用Docker 容器脚本)
  • Android 13.0 Settings主页面去掉FocusRecyclerView相关功能
  • Python(四)字符串
  • WPF中ElementName与RelativeSource绑定的局限性以及对策
  • 基于PHP语言的会员系统搭建(Docker版)