深度学习框架 PyTorch 安装步骤(技术)

简介: 深度学习框架 PyTorch 安装步骤(技术)

前言



去年双十一给自己买了台 8G 显卡内存的笔记本,之所以对显卡要求这么高是因为对人工智能感兴趣,想在自己笔记本上训练深度学习模型,因此一定要配一个好一点的 GPU。

666.webp.jpg


除了笔记本之外,前前后后陆陆续续也买了好些人工智能方面的书籍,刚工作一年时候也学习过这块,像《DEEP LEARNING 深度学习》、《TensorFLow实战》、《机器学习》、《图像识别应用开发》都是学习过的,也对跟着 PyTorch 官网教程走过,还有 github 上的《动手学深度学习》电子书,不过说实话,这个方向,挺难的,研究没出什么成果。但即便如此,遇到这块的好书还是会买着,给自己的兴趣一个随时能继续研究学习的环境。


555.webp.jpg

人工智能书籍


近期好朋友说参加了个关于人工智能学习的线下班,想起大约三年前一期学习过 python,而且本身就对这块感兴趣,还有这阵子不是很忙,有时间。那行吧,好机会,重新拾起来吧,工具(专门为深度学习买的笔记本)、资料(那些书籍)都有。真的很感激曾经的自己一直为兴趣而保留的环境,过去的自己就等待着这么个机会,待天时地利人和都齐了,随时出发!


正文


1.安装 Nvidia CUDA 并行计算工具包


官网下载地址:https://developer.nvidia.com/cuda-downloads

选择适合自己的版本下载,我选择的是,

Operating System: Windows

Architecture: x86_64

Version: 11

Installer Type: exe(network)


444.webp.jpg

cuda工具包下载界面


2.安装 Pytorch


官网地址:https://pytorch.org/get-started/locally

根据自己的开发环境,找到安装命令,我的安装命令是,



pip3 install --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cu113/torch_nightly.html

333.webp.jpg

pytorch 安装命令


3. 安装 PyTorch 包


我使用的开发工具是 PyCharm,属于最喜欢的 JetBrain 系列开发工具,新建一个工程(这里不要选择虚拟环境,少点坑,我最喜欢的 pipenv 都没用),执行安装命令等待安装完成即可。


11111.webp.jpg

等待安装完成


4.验证安装结果


一切顺利,显示安装完成。


77.webp.jpg

安装完成


接下来用代码验证一下,


import torch
x = torch.rand(5, 3)
print('PyTorch is ok? \n', x)
is_cuda_ok = torch.cuda.is_available()
print('cuda is ok? \n', is_cuda_ok)


可以看到 PyTorch 安装成功了,CUDA 并行计算工具包也正常工作了。


88.webp.jpg

代码验证


5.深度学习框架 PyTorch 环境搭建完毕,说点题外话


一般像这种环境搭建,可能不少人第一反应就是百度找一个教程,复制粘贴一下,这个流程走下来,一般情况下是能解决问题的。不过我这里给个建议:百度找的教程没走下来的话,去找官网教程,大多数是英文的,一定要静下心来慢慢看。

希望接下来的学习一切顺利咯!


目录
相关文章
|
6月前
|
边缘计算 人工智能 PyTorch
130_知识蒸馏技术:温度参数与损失函数设计 - 教师-学生模型的优化策略与PyTorch实现
随着大型语言模型(LLM)的规模不断增长,部署这些模型面临着巨大的计算和资源挑战。以DeepSeek-R1为例,其671B参数的规模即使经过INT4量化后,仍需要至少6张高端GPU才能运行,这对于大多数中小型企业和研究机构来说成本过高。知识蒸馏作为一种有效的模型压缩技术,通过将大型教师模型的知识迁移到小型学生模型中,在显著降低模型复杂度的同时保留核心性能,成为解决这一问题的关键技术之一。
548 6
|
6月前
|
机器学习/深度学习 数据采集 自然语言处理
29_序列标注技术详解:从HMM到深度学习
序列标注(Sequence Labeling)是自然语言处理(NLP)中的一项基础任务,其目标是为序列中的每个元素分配一个标签。在NLP领域,序列标注技术广泛应用于分词、词性标注、命名实体识别、情感分析等任务。
610 0
|
6月前
|
机器学习/深度学习 人工智能 PyTorch
PyTorch深度学习 ? 带你从入门到精通!!!
🌟 蒋星熠Jaxonic,深度学习探索者。三年深耕PyTorch,从基础到部署,分享模型构建、GPU加速、TorchScript优化及PyTorch 2.0新特性,助力AI开发者高效进阶。
PyTorch深度学习 ? 带你从入门到精通!!!
|
7月前
|
机器学习/深度学习 自然语言处理 PyTorch
Transformer自回归关键技术:掩码注意力原理与PyTorch完整实现
掩码注意力是生成模型的核心,通过上三角掩码限制模型仅关注当前及之前token,确保自回归因果性。相比BERT的双向注意力,它实现单向生成,是GPT等模型逐词预测的关键机制,核心仅需一步`masked_fill_`操作。
572 0
Transformer自回归关键技术:掩码注意力原理与PyTorch完整实现
|
7月前
|
机器学习/深度学习 存储 PyTorch
Neural ODE原理与PyTorch实现:深度学习模型的自适应深度调节
Neural ODE将神经网络与微分方程结合,用连续思维建模数据演化,突破传统离散层的限制,实现自适应深度与高效连续学习。
630 3
Neural ODE原理与PyTorch实现:深度学习模型的自适应深度调节
|
8月前
|
机器学习/深度学习 存储 人工智能
深度解析大模型压缩技术:搞懂深度学习中的减枝、量化、知识蒸馏
本文系统解析深度学习模型压缩三大核心技术:剪枝、量化与知识蒸馏,详解如何实现模型缩小16倍、推理加速4倍。涵盖技术原理、工程实践与组合策略,助力AI模型高效部署至边缘设备。
1553 2
|
9月前
|
机器学习/深度学习 存储 监控
基于深度学习YOLO框架的城市道路损伤检测与评估项目系统【附完整源码+数据集】
本项目基于深度学习的YOLO框架,成功实现了城市道路损伤的自动检测与评估。通过YOLOv8模型,我们能够高效地识别和分类路面裂缝、井盖移位、坑洼路面等常见的道路损伤类型。系统的核心优势在于其高效性和实时性,能够实时监控城市道路,自动标注损伤类型,并生成损伤评估报告。
506 0
基于深度学习YOLO框架的城市道路损伤检测与评估项目系统【附完整源码+数据集】
|
9月前
|
机器学习/深度学习 自动驾驶 算法
基于深度学习的YOLO框架的7种交通场景识别项目系统【附完整源码+数据集】
在智慧交通和智能驾驶日益普及的今天,准确识别复杂交通场景中的关键元素已成为自动驾驶系统的核心能力之一。传统的图像处理技术难以适应高动态、复杂天气、多目标密集的交通环境,而基于深度学习的目标检测算法,尤其是YOLO(You Only Look Once)系列,因其检测速度快、精度高、可部署性强等特点,在交通场景识别中占据了重要地位。
1054 0
基于深度学习的YOLO框架的7种交通场景识别项目系统【附完整源码+数据集】
|
10月前
|
机器学习/深度学习 存储 PyTorch
PyTorch + MLFlow 实战:从零构建可追踪的深度学习模型训练系统
本文通过使用 Kaggle 数据集训练情感分析模型的实例,详细演示了如何将 PyTorch 与 MLFlow 进行深度集成,实现完整的实验跟踪、模型记录和结果可复现性管理。文章将系统性地介绍训练代码的核心组件,展示指标和工件的记录方法,并提供 MLFlow UI 的详细界面截图。
428 2
PyTorch + MLFlow 实战:从零构建可追踪的深度学习模型训练系统
|
10月前
|
机器学习/深度学习 PyTorch 算法框架/工具
提升模型泛化能力:PyTorch的L1、L2、ElasticNet正则化技术深度解析与代码实现
本文将深入探讨L1、L2和ElasticNet正则化技术,重点关注其在PyTorch框架中的具体实现。关于这些技术的理论基础,建议读者参考相关理论文献以获得更深入的理解。
304 4
提升模型泛化能力:PyTorch的L1、L2、ElasticNet正则化技术深度解析与代码实现

推荐镜像

更多
下一篇
开通oss服务