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


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

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


目录
相关文章
|
16小时前
|
机器学习/深度学习 传感器 人工智能
基于深度学习的图像识别技术在自动驾驶系统中的应用
【5月更文挑战第15天】随着人工智能技术的飞速发展,尤其是深度学习在图像处理领域的突破性进展,自动驾驶系统得以实现更加精准和高效的环境感知。本文章深入探讨了深度学习技术在自动驾驶车辆图像识别中的运用,分析了卷积神经网络(CNN)等模型在车辆、行人以及交通标志识别上的优势与挑战。同时,本文还针对当前自动驾驶系统中存在的数据偏差、实时处理能力及安全性问题提出了潜在的解决策略,并展望了未来发展趋势。
|
17小时前
|
机器学习/深度学习 传感器 自动驾驶
基于深度学习的图像识别技术在自动驾驶系统中的应用
【5月更文挑战第14天】 随着人工智能技术的突飞猛进,特别是深度学习在图像处理领域的应用,为自动驾驶汽车的环境感知和决策提供了新的解决方案。本文将探讨基于深度神经网络的图像识别技术如何增强自动驾驶系统的性能,包括车辆检测、行人识别以及交通标志识别等方面。通过分析现有文献和最新研究成果,我们概述了关键技术挑战,并提出了潜在的改进方向。此外,文章还对目前自动驾驶领域中深度学习技术的实际应用情况进行了案例分析,以期为未来研究提供参考。
|
16小时前
|
机器学习/深度学习 传感器 自动驾驶
基于深度学习的图像识别技术在自动驾驶系统中的应用
【5月更文挑战第14天】 随着人工智能技术的飞速发展,深度学习已成为推动多个领域革新的关键力量。特别是在图像识别领域,深度学习技术已经实现了巨大的突破,为自动驾驶系统提供了强大的视觉处理能力。本文将探讨深度学习技术如何增强自动驾驶车辆的环境感知能力,以及这些技术是如何被集成到复杂的驾驶决策过程中的。通过对现有文献和最新研究成果的综合分析,我们将概述当前最前沿的算法和架构,并讨论它们在实际道路测试中的表现。
|
17小时前
|
机器学习/深度学习 传感器 自动驾驶
基于深度学习的图像识别技术在自动驾驶系统中的应用
【5月更文挑战第14天】 随着人工智能领域的飞速发展,特别是深度学习在图像处理和识别方面取得的重大进展,自动驾驶技术迎来了前所未有的发展机遇。本文旨在探讨基于深度学习的图像识别技术如何被集成到自动驾驶系统中,以及这种集成如何提高车辆的环境感知能力、决策效率和安全性。文中首先回顾了当前自动驾驶技术面临的挑战,随后详细介绍了深度学习技术的核心原理和在图像识别方面的应用,最后通过案例分析展示了该技术在实际自动驾驶场景中的表现和潜力。
17 7
|
17小时前
|
机器学习/深度学习 边缘计算 人工智能
揭秘深度学习在图像识别中的核心技术与挑战
【5月更文挑战第13天】 随着计算机视觉技术的飞速进步,深度学习已成为推动图像识别领域革新的主要动力。本文深入剖析了深度学习在图像识别任务中的核心算法和技术,包括卷积神经网络(CNN)的架构优化、数据增强的重要性、损失函数的选择以及模型压缩和加速策略。同时,文章也指出了当前技术面临的主要挑战,如数据集偏差、模型泛化能力、对抗性攻击等,并探讨了潜在的解决方案。通过综合分析和讨论,旨在为研究者和实践者提供指导,促进图像识别技术的发展和应用。
|
17小时前
|
机器学习/深度学习 PyTorch TensorFlow
Pytorch 与 Tensorflow:深度学习的主要区别(1)
Pytorch 与 Tensorflow:深度学习的主要区别(1)
15 2
|
17小时前
|
机器学习/深度学习 数据采集 监控
基于深度学习的图像识别技术在智能监控系统中的应用
【5月更文挑战第12天】 随着人工智能技术的迅猛发展,深度学习在图像处理领域已经取得了显著的成就。尤其是在图像识别任务中,卷积神经网络(CNN)等先进模型的应用极大提高了识别精度与效率。本文旨在探讨基于深度学习的图像识别技术如何被集成到智能监控系统中,以增强其自动检测和响应异常事件的能力。通过分析现有文献和最新研究成果,文章将详细阐述相关技术原理、系统架构、以及在实际场景中的应用效果,并讨论了面临的主要挑战和未来的发展方向。
|
17小时前
|
机器学习/深度学习 传感器 人工智能
基于深度学习的图像识别技术在自动驾驶系统中的应用
【5月更文挑战第12天】 随着人工智能技术的飞速发展,深度学习已成为推动多个科技领域前进的关键力量。特别是在图像处理和识别方面,深度学习模型已经展现出了超越传统算法的性能。本文将探讨一种基于深度卷积神经网络(CNN)的图像识别技术,并分析其在自动驾驶系统中的应用。我们将重点介绍该技术如何提高自动驾驶汽车对周围环境的感知能力,以及它在未来交通生态中的潜在影响。
|
17小时前
|
机器学习/深度学习 人工智能 算法
构建高效AI系统:深度学习优化技术解析
【5月更文挑战第12天】 随着人工智能技术的飞速发展,深度学习已成为推动创新的核心动力。本文将深入探讨在构建高效AI系统中,如何通过优化算法、调整网络结构及使用新型硬件资源等手段显著提升模型性能。我们将剖析先进的优化策略,如自适应学习率调整、梯度累积技巧以及正则化方法,并讨论其对模型训练稳定性和效率的影响。文中不仅提供理论分析,还结合实例说明如何在实际项目中应用这些优化技术。
|
17小时前
|
机器学习/深度学习 传感器 自动驾驶
探索基于深度学习的图像识别技术在自动驾驶领域的应用
【5月更文挑战第10天】 随着人工智能技术的飞速发展,特别是深度学习在图像处理和识别方面的突破性进展,自动驾驶汽车逐渐成为现实。本文将深入探讨深度学习技术在图像识别领域的应用,以及如何通过这些技术提高自动驾驶系统的性能和安全性。我们将分析卷积神经网络(CNN)在车辆检测、行人识别和交通标志识别中的作用,并讨论数据增强、迁移学习和对抗网络等策略如何帮助提升模型的泛化能力和鲁棒性。此外,文中还将涉及深度学习模型在实时处理和嵌入式系统部署时面临的挑战及其解决方案。