pytorch安装方法

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


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

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

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

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

其中Pytorch安装的路径在:

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


 

猜你喜欢

转载自blog.csdn.net/qq_34840129/article/details/88219664