Pytorch是facebook的开源深度学习框架,自推出以来,发展迅速。由于tensorflow1.x采用会话机制,不符合常规习惯,也给调试代码带来困难,作为对比,pytorch更符合部分人的审美和代码习惯。 tensorflow2.x也取悦大众,取消了会话机制。
Pytorch0.4.1版本开始支持windows系统,后续版本是1.0.x,1.1.x,1.2.x。然而从1.3.x,pytorch又没有windows版本,可谓特立独行(没有取悦大众)。另外,tensorflow,keras已经支持amd的显卡(ubuntu系统原生支持)。而 pytorch对于amd显卡的支持还不完善,需要ubuntu+docker,我们洋外洋科技也没有测试成功过。网上成功的按理也不多。2020/1/16日,pytorch发布了1.4.0版本,同样,没有windows版本,不过加入了python3.8的支持,torchvision后续也加入了python3.8支持。 而tensorflow那边,官方并没有加入python3.8的支持(我们提供了非官方的支持,仅限ubuntu系统,点击下载,但是很遗憾,windows系统下我们也没有编译成功,可能官方也没成功, 或许这是tensorflow没有官方支持python3.8的原因吧),keras倒是支持。另外mxnet因为是跨版本的whl,所以支持,我们猜测甚至3.9.x都支持。chainer也支持python3.8,但是官方宣布chainer将不再更新,官方转用pytorch。
趁着新鲜感,我们在ubuntu18.04上python3.8的虚拟环境里(py38)安装了cuda10和对应版本的pytorch,并简单测试了一下(点击下载测试代码)
关于我们
洋外洋科技: 洋外洋科技
分享快乐