百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 编程文章 > 正文

手把手教你搭建深度学习环境Pytorch版-Ubuntu

qiyuwang 2025-05-15 20:30 4 浏览 0 评论

引言

很多搞人工智能的小伙伴,刚开始学习,往往摸不着头脑怎么跑代码。跑代码的前提是要有个环境。本篇结合自己的亲身经历,带你搭建环境。

相关知识

  • Ubuntu是Linux系统的一种
  • 显卡驱动和cuda是两个不同的概念。驱动是直接和硬件打交道的,而cuda是在驱动之上用来完成加速计算的库。

方法

硬件环境:3090

操作系统:Ubuntu

第一步:装驱动

假设你拿到的是一台刚装完操作系统的电脑。首先我们需要安装显卡驱动。安装显卡驱动有两种方法,一种是简洁版,一种是繁琐版,需要官方下载驱动,好处就是可以连cuda一起装。

简洁版

点击作下角的9个点,选择附加驱动。

选择带Server Driver , server 专有 等关键词的。不要选tested ,open等。470支持的cuda版本低一些。大概是11.4,525可以支持12.0。 最后点击更改应用,重启计算机。完成。

繁琐版

【超详细】【ubunbu 22.04】 手把手教你安装nvidia驱动,有手就行,隔壁家的老太太都能安装_ubuntu安装nvidia显卡驱动-CSDN博客

看这个就可以了,过程中会让你选择要不要安装cuda。

最后将cuda添加到环境变量中

# 编辑配置文件,在最后添加以下内容
vim  ~/.bashrc
# 生效
source ~/.bashrc

将路径中cuda的版本换成你自己实际的版本。

检验自己是否安装成功

nvidia-smi

第二步:安装Anaconda

Anaconda是一个python包管理工具,可以创建不同的环境,针对不同的项目。

打开这链接

anaconda | 镜像站使用帮助 | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

点击连接,选择合适的版本,一般电脑选择86_64

下载后,打开终端执行安装命令, 根据提示安装即可

bash  Anaconda3-2023.09-0-Linux-x86_64.sh

更改镜像源,这是为了提高下载包的速度。通过修改用户目录下的 .condarc 文件来使用 TUNA 镜像源。

cd ~
vim .condara
# 删除之前的内容,添加以下
channels:
  - defaults
show_channel_urls: true
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch-lts: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  deepmodeling: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/
# 最后按ESC键,输入冒号wq, 保存退出。

# 生效
source ~/.bashrc

完成后你会看到前面多了一个base

第三步:创建环境

# --name 指定环境名称, python=3.8指定python的版本。
# 一般用python3.8,具体根据自己的项目选择
conda create --name myenv  python=3.8

# 激活环境
conda activate myenv

下面安装pytorch

浏览器搜素pytorch,打开pytorch官方或点击下方链接

PyTorch

我们以之前的版本为例,加入你一开始装的驱动是11.4, 那在这里我们就选择11.4,或则低于11.4的版本。此外还要选则用pip安装,因为其他命令会特别慢,有时下载过程中会断线。

然后你可以根据项目需要使用pip安装你需要python包。

结语

今天就分享到这里啦,如果你有任何问题,请私信我。

相关推荐

PPO 强化学习机械臂 IK 训练过程可视化利器 Tensorboard

视频讲解:PPO强化学习机械臂IK训练过程可视化利器Tensorboard_哔哩哔哩_bilibili...

免费定时运行Python程序并存储输出文档的服务推荐

免费定时运行Python程序并存储输出文档的服务推荐以下是几种可以免费定时运行Python程序并存储输出结果的云服务方案:1.PythonAnywhere特点:提供免费的Python托管环境...

实战搭建深度学习运行

一、准备环境ubuntu22.04nvidia显卡这里使用的是RTX3060已安装Python3.10二、安装pip3...

部署Django到阿里云服务器教程

基于Ubuntu16.04+Python3+nginx+mysql+Django接下来先安装这些必要的环境,这些操作都是在你已经购买了阿里云服务器。并且在本地你已经搭建好了Django...

Ubuntu 22.04安装MySQL : Qwen2.5 模型对话数据收集与微调教程

1安装MySQL首先来到(安装)根目录下,开启终端...

3D点云数据处理简明教程【Python】

近年来,在AR和空间计算技术的大力推动下,在Apple和Meta等大公司的支持下,最近推出了AppleVisionPro,3D理解领域受到越来越多的关注。...

Ubuntu22.04中定时执行任务(cron)设置

做这个实验,主要涉及内容如下:cronTest.py待执行的python文件,代码会读入本地文件内容,并打印当前时间cron_run.sh将执行cronTest.py的语句,包装成shellcro...

使用vllm部署DeepSeek

1、基础环境1.1、基础环境准备RTX3090*2卡ubuntu22.04python3.12cuda12.4pytorch2.5.1...

一键打包,随时运行,Python3项目虚拟环境一键整合包的制作(Venv)

之前我们介绍了如何使用嵌入式Python3环境给项目制作一键整合包,在使用嵌入式Python环境时,通常是作为另一个应用程序的一部分,而Python3虚拟环境是为了在开发过程中隔离项目所需的...

从零开始:Ubuntu Server中MySQL 8.0的安装与Django数据库配置详解

Ubuntu系统纯净安装MySQL8.01、安装Mysql8.0sudoaptinstallmysql-server...

同一台电脑如何共存多个版本Python?

同一台电脑可以安装多个版本Python吗?可以共存!当前电脑安装了Python3.8.832位,为了调用大漠插件dll,但是我又想安装较新版本的64位Python3.12.0,就涉及到多个Pytho...

我把 ML 模型编译成 C 后,速度竟提升了 1000 倍!

...

zabbix企业微信告警

zabbix企业微信告警的前提是用户有企业微信且创建了一个能够发送消息的应用,具体怎么创建可以协同用户侧企业微信的管理员。第一步:企业微信准备我们需要的内容包括企业ID,应用的AgentId和应用的S...

「2022 年」崔庆才 Python3 爬虫教程 - urllib 爬虫初体验

首先我们介绍一个Python库,叫做urllib,利用它我们可以实现HTTP请求的发送,而不用去关心HTTP协议本身甚至更低层的实现。我们只需要指定请求的URL、请求头、请求体等信息即...

手把手教你搭建深度学习环境Pytorch版-Ubuntu

引言很多搞人工智能的小伙伴,刚开始学习,往往摸不着头脑怎么跑代码。跑代码的前提是要有个环境。本篇结合自己的亲身经历,带你搭建环境。相关知识Ubuntu是Linux系统的一种...

取消回复欢迎 发表评论: