pytorch安装方法

本文详细介绍如何在Windows环境下,使用Anaconda和CUDA配置PyTorch环境,包括选择合适版本、conda安装步骤及验证安装成功的方法。同时,还介绍了如何安装torchvision,以便于后续深度学习项目的数据集和模型使用。
摘要由CSDN通过智能技术生成

 

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
--------------------- 


 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值