pytorch安装方法

 

1 、进入官网https://pytorch.org/

pytorch安装方法_第1张图片
找到适合自己的版本,CUDA、平台、Python等,然后会自动提示“Run this command” 命令指令进行安装。我的是Python3.5(Anaconda4.2)+CUDA8.0+windows10,当然如果电脑没有GPU,CUDA就选none。

2、conda 安装Pytorch包
Win+R进入windows运行,输入cmd进入命令行窗口,输入以上命令稍等片刻可以自动下载安装了,如下图所示。

pytorch安装方法_第2张图片

安装好后Pytorch后,还有个torchvision安装,这个主要集成了一些数据集,深度学习模型,一些转换等,以后需要使用还是很方便的。

pytorch安装方法_第3张图片

3、测试安装
进入Python环境测试安装的包是否成功,如下图所示。import torch导入没报错就说明安装成功,torch.cuda.is_available()是True就表示支持GPU啦~

 

其中Pytorch安装的路径在:

附:关于Pytorch+caffe2部署不错的例子:https://zhuanlan.zhihu.com/p/32342366
--------------------- 


 

你可能感兴趣的:(机器学习)