pytorch简介
PyTorch 是一个基于 Torch(LUA语言) 的 Python 开源机器学习库,用于自然语言处理等应用程序。 它主要由 Facebook 的人工智能研究小组开发。PyTorch是一个Python包,提供两个高级功能:
- 具有强大的GPU加速的张量计算(如NumPy)
- 包含自动求导系统的的深度神经网络
pytorch安装
使用pip install,官网安装太慢,选择清华源下载,执行指令如下:
pip install --upgrade torch torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple
等待安装成功。
进入python环境,查看pytorch版本:
import torch
print(torch.__version__)
安装成功。