PyTorch介绍与安装

介绍

Facebook的AI研究团队发布了Python工具包——PyTorch，专门用于GPU加速的深度学习编程，实现了在Python语言下使用Torch。

安装

PyTorch的安装非常方便。在主页中选择对应的操作系统、包管理器、Python版本、CUDA版本信息后会提示安装命令。不完全组合如下：

• Linux + pip + Python2.7 + CUDA7.5

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

• Linux + pip + Python2.7 + CUDA8.0

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

• Linux + pip + Python3.5 + CUDA7.5

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

• Linux + pip + Python3.5 + CUDA8.0

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

卸载

sudo pip uninstall torch