什么是pytorch?

PyTorch是一个基于python的科学计算包

为什么使用pytorch?

作为NumPy的替代品,可以利用GPU的性能进行计算
作为一个高灵活性、速度快的深度学习平台

安装

  • 配置python环境:3.6或者3.7都可以
  • 安装pip或者conda python管理器,个人觉得使用conda最好,方便管理环境
  • 使用命令行安装Pytorch:根据pytorch官网的提示安装
    比如我是linux系统 python3.6 CUBA10.0 pip 就可以
    Task1---pytorch入门
    根据提示,我只需要在命令行敲

pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.1.post2-cp36-cp36m-linux_x86_64.whl

pip3 install torchvision

pytorch基础概念:

相关文章: