PyTorch学习笔记之--环境搭建

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 
发布了29 篇原创文章 · 获赞 43 · 访问量 23万+

猜你喜欢

转载自blog.csdn.net/xiaoyuge16/article/details/81227779