PyTorch目前支持的环境
PyTorch官方网站是https://pytorch.org/
官网上面提供了对应的安装方法。目前已经有了Windows版本的PyTorch,这是令人十分开心的。但是Windows版的只支持Python3,不支持Python2,所以对于Windows用户,如果需要使用Python2,还是需要用Linux虚拟机或者双系统。
安装过程
安装过程相对来说还是很简单的。
可以使用anaconda安装,已经集成了很多包,而且有虚拟环境很好用。
之后,选择好对应的环境,如对于使用Linux、anaconda、Python2、并且不需要GPU的,官网给出了对应的安装命令:
conda install pytorch-cpu torchvision-cpu -c pytorch
安装过程提示各种安装,选择yes即可;
但是anaconda下载可能会比较慢,建议换成清华的源,如果还是出错或者网速慢,那么可以使用pip安装试试。
在官网选择pip安装,会出现如下格式的命令:
pip install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp27-cp27mu-linux_x86_64.whl
pip install torchvision
# if the above command does not work, then you have python 2.7 UCS2, use this command
pip install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp27-cp27m-linux_x86_64.whl
当然也可以首先自己下载http://download.pytorch.org/whl/cpu/torch-0.4.0-cp27-cp27mu-linux_x86_64.whl 到本地,然后在本地使用
pip install torch-0.4.0-cp27-cp27mu-linux_x86_64.whl
pip install torchvision