ubuntu系统搭建pytorch环境详细步骤【笔记】

embedded/2024/9/23 18:31:39/

实践设备:华硕FX-PRO(NVIDIA GeForce GTX 960M)

搭建PyTorch环境的详细步骤如下:

  1. 安装Ubuntu系统:

    • 下载Ubuntu的镜像文件并制作启动盘。
    • 将启动盘插入计算机,启动计算机并按照提示安装Ubuntu系统。
  2. 配置镜像源,安装必要环境:
    这里以添加清华大学的镜像源地址为例,具体细节请参考https://blog.csdn.net/qq_37592750/article/details/138258932。
    2.1.打开终端(Ctrl+Alt+T)。
    2.2.使用文本编辑器编辑sources.list文件。注释掉原有的源地址(在每一行前加上#)。
    2.3.添加清华大学的镜像源地址。以Ubuntu 20.04为例,可以添加以下内容:

deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ focal main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ focal-updates main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ focal-backports main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ focal-security main restricted universe multiverse

2.4.运行以下命令更新系统:

sudo apt update

3.安装NVIDIA驱动和CUDA工具:
(此步骤具体细节安装配置请参考《UEFI安全启动模式下安装Ubuntu的NVIDIA显卡驱动》):
image.png
image.png

4.安装Anaconda:
4.1.下载并安装Anaconda。
你可以从Anaconda的官方网站下载适合Linux系统的版本。
image.png

4.2.下载完成后,打开终端,导航到下载的文件所在的目录,
并执行以下命令来安装Anaconda:

bash Anaconda3-版本号-Linux-x86_64.sh

按照屏幕上的提示完成安装过程。
image.png
20240427212315.png
20240427212411.png

4.3.配置Anaconda环境:
安装完成后,需要配置Anaconda环境。
打开终端,并编辑 ~/.bashrc 文件: vim ~/.bashrc
在文件的末尾添加以下内容:

export PATH="/home/用户名称/anaconda3/bin:$PATH"

image.png
保存并关闭文件后,运行以下命令使更改生效:source ~/.bashrc然后,测试Anaconda是否安装成功:conda --version如果显示版本信息,则表示Anaconda安装成功。
在这里插入图片描述

  1. 安装Python和pip:
    • 终端中运行以下命令安装Python和pip:
sudo apt install python3 python3-pip
  1. 安装PyTorch依赖库:
    • 终端中运行以下命令安装PyTorch所需的依赖库:
sudo apt install libopenblas-dev libblas-dev m4 cmake cython python3-dev python3-yaml python3-setuptools

7.配置pip源(这一步很重要,影响下一步骤操作的下载速度
7.1.创建或修改pip配置文件:
你可以在用户的家目录下创建或修改一个pip配置文件:

mkdir -p ~/.config/pip
vim ~/.config/pip/pip.conf

7.2.添加pip源:
在pip.conf文件中,你可以选择一个国内的pip源,如阿里云、清华大学等,
阿里云pip源的示例:

[global]
index-url = https://mirrors.aliyun.com/pypi/simple

清华大学示例:

[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple

保存并关闭文件。之后,当你使用pip安装包时,pip会自动使用配置好的源。
在这里插入图片描述
image.png

  1. 安装PyTorch:
    image.png
    终端中运行以下命令安装PyTorch:
pip3 install torch torchvision torchaudio

image.png

  1. 验证安装:
    • 在终端中运行以下命令验证PyTorch是否成功安装:
(base) ubuntu@GL552VW:~$ python3
Python 3.11.7 (main, Dec 15 2023, 18:12:31) [GCC 11.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
True
>>> print(torch.__version__)
2.3.0+cu121
>>>

在这里插入图片描述
以上是在Ubuntu系统上搭建PyTorch环境的详细步骤。

申明: 
所有资料为本人收集整理,仅限个人学习使用,勿做商业用途。


http://www.ppmy.cn/embedded/20381.html

相关文章

Pytorch 的实际应用 学习笔记

一. 模型的下载 weights为false时则为没有提前经过训练的模型,为true时则经过了提前训练 vgg16_false torchvision.models.vgg16(weightsFalse) vgg16_true torchvision.models.vgg16(weightsTrue) 打印 二. 模型的修改 (1)添加操作 …

Unity类银河恶魔城学习记录15-5,6 p157 Audio time limiter p158 Area sound

Alex教程每一P的教程原代码加上我自己的理解初步理解写的注释,可供学习Alex教程的人参考 此代码仅为较上一P有所改变的代码 【Unity教程】从0编程制作类银河恶魔城游戏_哔哩哔哩_bilibili​​ AreaSound.cs using System.Collections; using System.Collections.G…

1146. 快照数组

java版本 class SnapshotArray {int id 0;List<int[]>[] snapshots;public SnapshotArray(int length) {snapshots new List[length];for (int i 0; i < length; i) {snapshots[i] new ArrayList<int[]>();}}public void set(int index, int val) {snapsho…

Github使用指南

天行健&#xff0c;君子以自强不息&#xff1b;地势坤&#xff0c;君子以厚德载物。 每个人都有惰性&#xff0c;但不断学习是好好生活的根本&#xff0c;共勉&#xff01; 文章均为学习整理笔记&#xff0c;分享记录为主&#xff0c;如有错误请指正&#xff0c;共同学习进步。…

Bun 入门到精通(二)——初始化

bun init 使用 bun init 可以搭建一个新项目&#xff0c;类似于 npm init&#xff0c;同样&#xff0c;bun init 也可以接收 -y/--yes 的参数&#xff0c;同样类似于 npm init -y。 bun init工作原理 先创建一些默认文件 包文件&#xff1a;带 name 属性的 package.json语言…

机器学习:深入解析SVM的核心概念(问题与解答篇)【一、间隔与支持向量】

直接阅读原始论文可能有点难和复杂&#xff0c;所以导师直接推荐我阅读周志华的《西瓜书》&#xff01;&#xff01;然后仔细阅读其中的第六章&#xff1a;支持向量机 间隔与支持向量 问题一&#xff1a;什么叫法向量&#xff1f;为什么是叫法向量 在这个线性方程中&#xff…

js动态设置css主题(Style-setProperty)

hex颜色转RGB hex2Rgb(str) {str str.replace("#", "");const hxs str.match(/../g);for (let index 0; index < 3; index) hxs[index] parseInt(hxs[index], 16);return hxs; } RGB转HXS rgb2hex(r,g,b){const hexs [r.toString(16), g.toString…

计算机网络—网络层

一、网络层的功能 网络层的任务 主要任务是通过路由器实现网络互联&#xff0c;进而实现数据报在各网络之间的传输 “数据报”就是从数据链路层接收的分组好的数据&#xff0c;又叫“IP数据报”或“IP分组” 网络层提供的两种服务 1面向连接的虚电路服务 当两台计算机进行…