进入官网:https://pytorch.org/
根据配置可以生成相应的安装命令,运行之后,会安装torch和torchvision这两个库。
torch 是主模块, 用来搭建神经网络;
torchvision 是辅模块, 有数据库, 还有一些已经训练好的神经网络可以直接使用 , 比如 (VGG, AlexNet, ResNet)
pytorch与numpy
pytorch相当于神经网络中的numpy库,pytorch与numpy具有很好的兼容性,pytorch中的tensor与numpy中的array能自由转换。
-