PyTorch是Facebook团队于2017年1月发布的一个深度学习框架。PyTorch采用Python语言接口来实现编程,就像带GPU的NumPy,与Python 一样属于动态框架。PyTorch继承了Torch灵活、动态的编程环境和用户友好等特点,支持以快速与灵活的方式构建动态神经网络,还允许在训练过程中快速更改代码而不妨碍其性能,支持动态图形等尖端AI模型的功能。
为何选择PyTorch
PyTorch是一个建立在Torch库之上的Python包,旨在加速深度学习应用。它提供一种类似NumPy的抽象方法来表征张量(或多维数组),可以利用GPU来加速训练。PyTorch采用了动态计算图(Dynamic Computational Graph)结构,是基于tape的autograd系统的深度神经网络。其他很多框架采用静态计算图。通过PyTorch的一种称为反向模式自动微分(Reverse-Mode Auto Differentiation)的技术,我们可以非常方便地构建网络。
torch是PyTorch中的一个重要包,它包含了多维张量的数据结构以及基于其上的多种数学操作。
PyTorch由4个主要包组成:
1、torch:类似于NumPy的通用数组库,可将张量类型转换为torch.cuda.TensorFloat,并在GPU上进行计算。
2、torch.autograd:用于构建计算图形并自动获取梯度的包。
3、torch.nn:具有共享层和损失函数的神经网络库。
4、torch.optim:具有通用优化算法(如SGD、Adam等)的优化包。
PyTorch(GPU版)的安装配置
总的思路:硬件 到 软件,最后 再验证。
1、英伟达显卡
GPU(Graphics Processing Unit, 图像处理单元):主要用于在屏幕上显示图像,用于与视频、图像处理相关的任务。分为 独立显卡 和 核显。
可以通过360驱动大师查看硬件配置信息:
如下图所示:此电脑的显卡型号为NVIDIA GeForce RTX 3060 Laptop GPU(6GB/蓝天(CLEVO))
2、显卡驱动
驱动的主要作用是让计算机识别特定的硬件。
官方下载驱动链接:官方驱动 | NVIDIA
注:复制粘贴到迅雷中下载会更快。
3、下载Anaconda
1)Python环境
2)Conda命令,安装包:conda install 包名
3)虚拟环境:创建多个相互独立的虚拟环境,需要用到哪个环境就进入到哪个环境。
Anaconda历史版本下载: https://repo.continuum.io/archive/
4、安装Anaconda
5、创建并验证虚拟环境
创建虚拟环境:+ 指定名称 + 指定python版本号
conda create -n
打开虚拟环境列表:
conda env list
创建虚拟环境: + 指定名称 +指定python版本号
conda create -n
打开虚拟环境列表:
conda env list
进入指定的虚拟环境:+ 虚拟环境名称
conda activate
检查当前虚拟环境的安装包:
conda list
退出当前的虚拟环境:
conda deactivate
6、安装PyTorch
PyTorch官网链接:PyTorch
进入到指定的虚拟环境(这里我选择:zwb_test01)中安装PyTorch:
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
该虚拟环境中的安装包列表:
conda list
7、验证PyTorch CUDA ucDNN
导入torch包:
import torch
验证CUDA是否安装成功:
torch.cuda.is_available()
显示CUDA的版本号:
torch.version.cuda
验证CUDNN是否安装成功:
torch.backends.cudnn.is_available()
显示CUDNN的版本号:
torch.backends.cudnn.version()
8、安装PyCharm
1)能够进行代码补全
2)代码运行,一键完成
3)实时语法错误检查
4)配置一个项目的时候,需要配置好Python解释器
PyCharm的历史版本的下载:https://www.jetbrains.com/pycharm/download/previous.html
9、激活PyCharm
激活码链接:微信扫一扫
打开PyCharm的激活码:全选(Ctrl + A) 后 复制(Ctrl + C)
Jupyter Notebook
创建密码
打开一个文件夹:鼠标右键,选择:在终端打开。
1、生成配置文件
jupyter notebook --generate-config
2、在该配置文件所在的文件夹下,鼠标右键,选择:在终端打开。
ipython
from notebook.auth import passwd
passwd()
输入/验证密码后,需要复制字符串到一个文本文档中,等会需要复制粘贴到配置文件中。
exit()
在一个文件夹下,鼠标右键,选择:在终端打开。
jupyter notebook
密码设置成功: