pytorch 安装

torch 作为和tensorflow齐名的深度学习框架,并且与tensorflow不一样的动态网络图,tensorflow是静态的,pytorch作为python版本的torch真是越用越顺手,而且有facebook团队专门维护,开源至今1年多的时间圈粉无数。我也计划主要的深度学习,强化学习的大部分工作渐渐由tensorflow迁移到pytorch。

如果是在ubuntu或者osx上安装,参见官网,自己选择os和python版本以及是否有cuda,选择好以后copy其代码就行:pytorch

pytorch目前官方没有windows版本放出,如果想在windows上玩pytorch怎么办?我们有conda呀。。。。

# for CPU only packages
conda install -c peterjc123 pytorch-cpu

# for Windows 10 and Windows Server 2016, CUDA 8
conda install -c peterjc123 pytorch

# for Windows 10 and Windows Server 2016, CUDA 9
conda install -c peterjc123 pytorch cuda90

# for Windows 7/8/8.1 and Windows Server 2008/2012, CUDA 8
conda install -c peterjc123 pytorch_legacy

另外torchvision的安装:

 pip install torchvision

参考1
参考2

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容