win10+pytorch+cuda安装

本文介绍了如何在Windows 10系统中安装PyTorch和CUDA。首先,通过NVIDIA控制面板检查CUDA版本,然后从PyTorch官网下载适合的版本,并根据生成的命令进行安装。最后,通过运行特定命令验证PyTorch安装是否成功。
摘要由CSDN通过智能技术生成

1、查看cuda版本

打开nvidia控制面板,点击系统信息-->组件-->查看NVCUDA.DLL那一项。可以看到我的cuda是10.1版本的。如果没有的话,可以先去安装cuda,这里不再介绍cuda的安装方法。

2、下载pytorch。官网链接https://pytorch.org/。根据自己的具体情况进行选择,会生成一个相应的安装命令

3、用上一步自己选择后生成的命令进行安装

4、判断是否安装成功pytorch。如果输入以下命令,无报错,且有输出,则证明pytorch安装成功。

import torch   
a = torch.Tensor([1.])   
a.cuda() 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值