什么是pytorch?
PyTorch是一个基于python的科学计算包
为什么使用pytorch?
作为NumPy的替代品,可以利用GPU的性能进行计算
作为一个高灵活性、速度快的深度学习平台
安装
- 配置python环境:3.6或者3.7都可以
- 安装pip或者conda python管理器,个人觉得使用conda最好,方便管理环境
- 使用命令行安装Pytorch:根据pytorch官网的提示安装
比如我是linux系统 python3.6 CUBA10.0 pip 就可以
根据提示,我只需要在命令行敲
pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.1.post2-cp36-cp36m-linux_x86_64.whl
pip3 install torchvision
pytorch基础概念:
- torch:包 torch 包含了多维张量的数据结构以及基于其上的多种数学操作。
- 张量 Tensors:PyTorch 提供了 CPU 张量和 GPU 张量,并且极大地加速了计算的速度。
参考网址:https://blog.csdn.net/tMb8Z9Vdm66wH68VX1/article/details/80682109
- 张量 Tensors:PyTorch 提供了 CPU 张量和 GPU 张量,并且极大地加速了计算的速度。