PyTorch是啥
PyTorch官方对于PyTorch的定位为:
一个使用GPU加速的numpy替换库
一个深度学习研究平台,提高最大灵活度和速度
具体点来讲, PyTorch是一个Python包,是Torch在Python上的衍生,原先的Torch是用Lua语言写的,虽然效率高,但是普及度不够,社区不够大,改成Python后,受众范围广泛了许多。并且有FaceBook这样的巨头支持,具有很好的发展前景。
安装
由于Lubuntu是基于Ubuntu衍生而来,因此安装过程可以参照Ubuntu下安装Pytorch的过程,一般有三种方式:基于conda安装;基于pip文件安装;基于源码编译。
基于conda安装
1. 安装conda命令(Anaconda或者Miniconda)
Anaconda包括包管理器和很多常用package,因此如果很多库你都还没有安装,直接安装Anaconda是不错的办法.
反之,如果很多依赖package你都安装好了,那么你安装miniconda 就好,也就是仅仅安装一个package管理器.
通常Anaconda的安装包都是.sh格式,假设是filename.sh,那么安装是在前面采用sh命令就可以进行安装了
sh filename.sh
2.修改conda镜像源
在终端中运行以下命令修改镜像源,可以明显加速安装.
# 优先使用清华conda镜像
conda config--prepend channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
# 也可选用科大conda镜像
conda config--prepend channels http://mirrors.ustc.edu.cn/anaconda/pkgs/free/
要查看镜像源是否安装成功的话,建议终端中运行以下命令:
conda config --set show_channel_urls yes
会生成一个~/.condarc文件,运行cat命令查看文件内容
cat ~/.condarc
显示内容为:
channels:- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
- http://mirrors.ustc.edu.cn/anaconda/pkgs/free/
-defaults
show_channel_urls:true
3. 安装Pytorch
我安装的方法是conda+python3.6+CPU,所以安装命令如下:
conda install pytorch-cpu torchvision -c pytorch
基于pip文件安装
1. 配置镜像源
请参考博文Ubuntu16.10下配置pip国内镜像源加速安装.
这一步是为加速安装torch vision进行的.
2. 下载.whl安装文件
3. 安装pytorch
用pip install .whl文件即可
pip install torch-0.3.0.post4-cp27-cp27mu-linux_x86_64.whl
pip install torchvision
基于源码编译
1.首先安装各种依赖
export CMAKE_PREFIX_PATH="$(dirname $(which conda))/../"# [anaconda root directory]
# Install basic dependencies
conda install numpy pyyaml mkl mkl-include setuptools cmake cffi typing
conda install-c mingfeima mkldnn
# Add LAPACK supportforthe GPU
conda install-c pytorch magma-cuda80 # or magma-cuda90 if CUDA 9
2.git代码
git clone --recursive https://github.com/pytorch/pytorch
cd pytorch
3. 安装pytorch
python setup.py install
整个安装过程会很漫长,请耐心等待。。。
测试
在终端中按以下顺序运行命令:
Python
import torch
参考链接: