引言¶
之前操作过torch
,是一个lua
编写的深度学习训练框架,后来facebook
发布了pytorch
,使用python
语言进行开发
pytorch
是在torch
的基础上发展而来的,它继承了许多内容,包括各种包的命名和类的定义,比如张量(tensor
)
参考:pytorch
目标¶
- 替代
NumPY
进行GPU
的运算 - 提供最大灵活性和速度的深度学习平台
安装¶
指定版本/操作系统/安装方式/python
语言/cuda
版本
当前配置:
- PyTorch Stable(1.0)
- Ubuntu 16.04
- Anacodna3
- Python 3.6
- CUDA 10.0
安装命令如下:
$ conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
加载torch
¶
命令行方式
$ python
Python 3.6.8 |Anaconda, Inc.| (default, Dec 30 2018, 01:22:34)
[GCC 7.3.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.__version__
'1.0.1.post2'
>>>
文件方式
from __future__ import print_function
import torch