深度学习框架 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 环境搭建完毕,说点题外话


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

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


目录
相关文章
|
29天前
|
机器学习/深度学习 存储 人工智能
MNN:阿里开源的轻量级深度学习推理框架,支持在移动端等多种终端上运行,兼容主流的模型格式
MNN 是阿里巴巴开源的轻量级深度学习推理框架,支持多种设备和主流模型格式,具备高性能和易用性,适用于移动端、服务器和嵌入式设备。
124 18
MNN:阿里开源的轻量级深度学习推理框架,支持在移动端等多种终端上运行,兼容主流的模型格式
|
26天前
|
机器学习/深度学习 数据可视化 PyTorch
PyTorch FlexAttention技术实践:基于BlockMask实现因果注意力与变长序列处理
本文介绍了如何使用PyTorch 2.5及以上版本中的FlexAttention和BlockMask功能,实现因果注意力机制与填充输入的处理。通过attention-gym仓库安装相关工具,并详细展示了MultiheadFlexAttention类的实现,包括前向传播函数、因果掩码和填充掩码的生成方法。实验设置部分演示了如何组合这两种掩码并应用于多头注意力模块,最终通过可视化工具验证了实现的正确性。该方法适用于处理变长序列和屏蔽未来信息的任务。
72 17
|
2月前
|
机器学习/深度学习 人工智能 自然语言处理
深入理解人工智能中的深度学习技术及其最新进展
深入理解人工智能中的深度学习技术及其最新进展
316 33
|
2月前
|
机器学习/深度学习 人工智能 自然语言处理
深入理解人工智能中的深度学习技术及其最新进展
深入理解人工智能中的深度学习技术及其最新进展
115 14
|
1月前
|
机器学习/深度学习 传感器 边缘计算
基于深度学习的图像识别技术在自动驾驶中的应用####
随着人工智能技术的飞速发展,深度学习已成为推动自动驾驶技术突破的关键力量之一。本文深入探讨了深度学习算法,特别是卷积神经网络(CNN)在图像识别领域的创新应用,以及这些技术如何被集成到自动驾驶汽车的视觉系统中,实现对复杂道路环境的实时感知与理解,从而提升驾驶的安全性和效率。通过分析当前技术的最前沿进展、面临的挑战及未来趋势,本文旨在为读者提供一个全面而深入的视角,理解深度学习如何塑造自动驾驶的未来。 ####
138 1
|
2月前
|
机器学习/深度学习 数据采集 传感器
基于深度学习的图像识别技术在自动驾驶中的应用研究####
本文旨在探讨深度学习技术,特别是卷积神经网络(CNN)在自动驾驶车辆图像识别领域的应用与进展。通过分析当前自动驾驶技术面临的挑战,详细介绍了深度学习模型如何提升环境感知能力,重点阐述了数据预处理、网络架构设计、训练策略及优化方法,并展望了未来发展趋势。 ####
153 6
|
1月前
|
机器学习/深度学习 算法框架/工具 网络架构
深度学习中的正则化技术及其对模型性能的影响
本文深入探讨了深度学习领域中正则化技术的重要性,通过分析L1、L2以及Dropout等常见正则化方法,揭示了它们如何帮助防止过拟合,提升模型的泛化能力。文章还讨论了正则化在不同类型的神经网络中的应用,并指出了选择合适正则化策略的关键因素。通过实例和代码片段,本文旨在为读者提供关于如何在实际问题中有效应用正则化技术的深刻见解。
|
2月前
|
机器学习/深度学习 监控 自动驾驶
基于深度学习的图像识别技术研究进展###
本文旨在探讨深度学习在图像识别领域的最新研究进展,重点分析卷积神经网络(CNN)的技术创新、优化策略及其在实际应用中的成效。通过综述当前主流算法结构、损失函数设计及数据集增强技巧,本文揭示了提升模型性能的关键因素,并展望了未来发展趋势。尽管未直接涉及传统摘要中的研究背景、方法、结果与结论等要素,但通过对关键技术点的深度剖析,为读者提供了对领域现状与前沿动态的全面理解。 ###
|
1月前
|
机器学习/深度学习 存储 人工智能
探索深度学习的奥秘:从理论到实践的技术感悟
本文深入探讨了深度学习技术的核心原理、发展历程以及在实际应用中的体验与挑战。不同于常规摘要,本文旨在通过作者个人的技术实践经历,为读者揭示深度学习领域的复杂性与魅力,同时提供一些实用的技术见解和解决策略。
49 0
|
1月前
|
机器学习/深度学习 传感器 自动驾驶
基于深度学习的图像识别技术及其在自动驾驶中的应用####
本文深入探讨了深度学习驱动下的图像识别技术,特别是在自动驾驶领域的革新应用。不同于传统摘要的概述方式,本节将直接以“深度学习”与“图像识别”的技术融合为起点,简述其在提升自动驾驶系统环境感知能力方面的核心作用,随后快速过渡到自动驾驶的具体应用场景,强调这一技术组合如何成为推动自动驾驶从实验室走向市场的关键力量。 ####
95 0

热门文章

最新文章