Pytorch 安装教程

前期环境

win7 64位

anaconda3.4.1

cuda9.0

1、进入pytorch官网(https://pytorch.org),根据自己的配置选择,如下图所示,然后打开dos窗口,在窗口中依次输入如下命令:

pip3 install http://download.pytorch.org/whl/cu90/torch-0.4.0-cp35-cp35m-win_amd64.whl 

pip3 install torchvision

 

 

2、输入第一条命令后会提示正在下载所需的包,等待下载完成即可

全部下载完后后,如果提示Successfully installed torch-0.4.0,则输入下一条命令:

pip3 install torchvision

 

3、执行完后提示Successfully installed pillow-5.2.0 torchvision-0.2.1,即安装完毕。

 

4、检查是否安装好pytorch,torchvision和pytorch是否支持cuda,先在dos窗口中输入Python,然后:

输入如下命令:

>>import torch

>>import torchvision

>>print(torch.cuda.is_available())

如果输出为True,则证明已经配置成功。

 

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值