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

Ubuntu系统安装NVIDIA驱动、CUDA、PyTorch等GPU深度学习环境

学习目标:

  • 在Ubuntu系统上安装CUDA、PyTorch等GPU深度学习环境,主要目标是为深入研究深度学习和深度强化学习提供高效的计算支持。
  • 通过构建GPU环境,计划掌握深度学习的基本概念和算法应用,提高模型训练效率,特别是在复杂项目中的应用能力。希望通过实践,增强对深度学习技术的理解和实际操作能力,为未来的研究奠定基础。
  • Anaconda是一个开源的Python和R语言分发版,专注于科学计算和数据分析。它提供包管理和环境管理工具(conda),方便用户创建虚拟环境,简化了库的安装和管理。

学习内容:

1 使用Anaconda工具创建单独任务(类任务)虚拟环境。

1.1 官网下载Linux安装文件。下载链接: 点击跳转

在这里插入图片描述

1.2 安装Anaconda。进入文件下载位置,文件夹空白处右键打开终端,终端输入以下命令安装,后面一直Yes即可。

在这里插入图片描述

# 给文件赋予执行权限并执行安装脚本
sudo chmod +x Anaconda3-2024.10-1-Linux-x86_64.sh
./Anaconda3-2024.10-1-Linux-x86_64.sh

1.3 如果在每次启动终端时,自动激活conda的base环境,需要取消其自动进入,可如下设置。

# 终端输入
conda config --set auto_activate_base false

1.4 如果终端显示“conda: command not found”,可能是因为conda没有正确添加到你的PATH环境变量中。

# 终端输入(法1export PATH="/home/jack/anaconda3/bin:$PATH"
source ~/.bashrc
# 终端输入(法2)
sudo gedit ~/.bashrc
#在文件的末尾添加以下行,保存退出
export PATH="/home/jack/anaconda3/bin:$PATH" 
#使更改生效
source ~/.bashrc
#检查PATH是否已更新
echo $PATH
#检查conda版本
conda --version

1.5 在conda中创建自己的虚拟环境。

# 终端输入
#查看当前安装的包和版本
conda list
#创建新环境 环境名称为jack、虚拟环境python版本为3.9,都可自主设计(注意兼容性),等待创建完成。
conda create --name jack python=3.9
#激活虚拟环境
conda activate jack

2 安装NVIDIA驱动

2.1 查看显卡及驱动信息

# 终端输入
#查看你的GPU型号
lspci | grep -i nvidia
# 查看NVIDIA驱动
nvidia-smi
# 查看CUDA版本(如果已安装)
nvcc --version

输出结果如图例所示:

在这里插入图片描述

2.2 安装推荐的 NVIDIA 驱动

# 终端输入 
sudo ubuntu-drivers autoinstall
# 安装完成后,重启计算机
sudo reboot
# 查看NVIDIA驱动
nvidia-smi
#使更改生效
source ~/.bashrc

2.3 安装CUDA Toolkit

# 终端输入 
sudo apt install nvidia-cuda-toolkit
# 安装完成后,重新启动终端检查驱动和CUDA是否正常
nvidia-smi
nvcc --version
# 配置环境变量
export PATH=/usr/local/cuda/bin:$PATH
#使更改生效
source ~/.bashrc

2.4 安装Pytorch

# jack环境下终端输入
# cu115是 nvcc --version 中查出的CUDA版本,必须保持一致
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu115

2.5 使用 PyTorch 测试 GPU 是否可用

# jack环境下终端输入
python
# 在代码输入区输入  按一次回车
import torch
# 检查当前系统中是否有可用的 GPU  
print("GPU available:", torch.cuda.is_available())
# 返回可用的 GPU 数量
print("GPU count:", torch.cuda.device_count())
# 返回当前正在使用的 GPU 的索引(从 0 开始) 
print("Current GPU:", torch.cuda.current_device())

验证安装成功时,输出结果如图例所示:
在这里插入图片描述


小结:

  • 在本次学习过程中,我们讨论了在 Ubuntu 系统上安装 CUDA、PyTorch 等 GPU 深度学习环境的过程。
  • 通过命令 nvidia-smi 和 nvcc --version,我们可以检查 GPU 驱动和 CUDA 版本。
  • 使用 PyTorch 的 torch.cuda 模块,可以验证 GPU 的可用性、数量以及当前使用的 GPU。
  • 这些步骤对于深度学习和深度强化学习项目至关重要,确保系统具备充分的计算能力以支持相关算法的训练与应用。

欢迎大家评论留言!!!


友情提示:

  • 专栏:深度学习(DL)
  • 专栏:深度强化学习(DRL)

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

相关文章:

  • 电子学会2024年3月青少年软件编程(图形化)等级考试试卷(三级)真题,含答案解析
  • 初学者指南:用例图——开启您的软件工程之旅
  • 二叉树遍历/算法数据结构
  • C#字符串的不可变性:内存管理与线程安全的优势分析
  • 【杂记】之语法学习第四课手写函数与结构体
  • 细说STM32单片机USART中断收发RTC实时时间并改善其鲁棒性的另一种方法
  • python使用turtle画图快速入门,轻松完成作业练习
  • 【C++】新手入门指南
  • C++使用开源ConcurrentQueue库处理自定义业务数据类
  • 在vue3的vite网络请求报错 [vite] http proxy error:
  • ElasticSearch 简单的查询。查询存在该字段的资源,更新,统计
  • FOFA使用教程之从零到精通
  • 【提高篇】3.2 GPIO(二,基本结构)
  • UE hard/soft reference| DDX DDY | Unity pcg color
  • macOS 应用公证指南:使用 fastlane 实现自动化公证流程
  • 深度学习:解密图像、音频和视频数据的“理解”之道20241105
  • uniapp 实现瀑布流
  • 计算机毕业设计 | springboot+vue智慧工地管理系统 前后端分离后台管理(附源码+文档)
  • vue中html如何转成pdf下载,pdf转base64,忽略某个元素渲染在pdf中,方法封装
  • Ubuntu下如何管理多个ssh密钥
  • [vulnhub] DarkHole: 1
  • 商淘云连锁企业管理五大功能 收银系统助力门店进销存同步
  • 统信UOS开发环境支持Perl
  • Stable Diffusion Web UI - ControlNet 姿势控制 openpose
  • java中Json字符串转换
  • springboot处理跨域请求
  • S32G-VNP-RDB2开发环境搭建
  • 分布式唯一ID生成(二): leaf
  • 【开发工具】Git
  • 【go从零单排】结构嵌套struct embedding