Linux下便捷安装 Pytorch命令指南

来源:互联网 发布:第三方支付平台 知乎 编辑:程序博客网 时间:2024/06/05 08:20

Pytorch是Facebook的开源深度学习框架,目前还不支持Windows系统,仅支持Linux和Mac。不过不要紧,深度学习的从业者多数习惯在Linux环境下进行相关的研究、开发。Pytorch的安装还是很简单的,官网的安装指导简单明了:点击选择按钮,生成安装命令。但问题是,有时候需要翻墙。这种安装方式的另外一个缺点是,源码被封装成 .so文件,看不到内部实现细节。因此,最好的安装方式其实是从GitHub下载源码,然后在一步步安装。这种安装方式的问题是需要事先安装各种依赖库,安装过程很不“傻瓜”,因此有了本博文,写给那些既不想手动从源代码安装,又懒得爬墙的童鞋。

方式一、Python2.7,无CUDA


1.pip installhttp://download.pytorch.org/whl/cu75/torch-0.1.12.post2-cp27-none-linux_x86_64.whl 

2.pip install torchvision

方式二、Python 2.7,CUDA 8.0


1.pip install http://download.pytorch.org/whl/cu80/torch-0.1.12.post2-cp27-none-linux_x86_64.whl 
2.pip install torchvision

方式三、Python 3.5, 无CUDA


1.pip install http://download.pytorch.org/whl/cu75/torch-0.1.12.post2-cp35-cp35m-linux_x86_64.whl 
2.pip install torchvision


方式四、Python 3.5,CUDA 8.0


1.pip install http://download.pytorch.org/whl/cu80/torch-0.1.12.post2-cp35-cp35m-linux_x86_64.whl 
2.pip install torchvision

原创粉丝点击