pytorch学习笔记（一）pytorch环境搭建

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__)

安装成功。