PyTorch是一个快速,灵活的实验深度学习框架[1]。下面介绍Ubuntu16.04 系统及Anaconda环境下,对于pytorch库(cpu版)的安装与demo测试。
系统环境:
- Ubuntu16.04
- Anaconda2-5.2.0-Linux-x86_64 [2]
pytorch 安装及环境配置
1、源码下载
git clone --recursive https://github.com/pytorch/pytorch
2、安装
- 打开,并进入刚下载的源码文件夹
cd ./pytorch
- 执行安装文件 setup.py
python setup.py install
注意:执行该命令时,可能会提示权限问题,你可使用chmod命令[3]将该问题解决。例如我的pytorch源码文件夹路径是:~/Downloads/pytorch。通过执行以下命令即可解决权限问题:
chmod 777 ~/Downloads/pytorch
如需了解更多有关chmod命令,可以参考文档[1]。
- pytorch包导入测试
在Python环境下输入命令:
import torch