确保Pytorch安装成功的一份不完全踩坑指南(windows10)

踩了不少坑,给大家po一个合理的安装顺序:

安装Anaconda

最好先在anaconda prompt里new一个conda的python3.6虚拟环境(方法去别的文章搜一下),再整理一下conda的pkg文件夹确保你以前没有安装过跟torch以及cuda toolkit相关的包。如果有的话,建议重装一下anaconda,或者clean一下list里的包。因为这个我配环境配了好久,出了很多问题。

CUDA Toolkit

先更新你的显卡驱动,现在nvidia的最新驱动好像是集合了,适配多张显卡的,更新完毕后。
打开prompt,查看CUDA适配的最高版本

nvidia-smi

如图,低于496.13以及11.5的cuda版本都是可以安装的。

然后先去pytorch官网上找你要安装的torch版本
如图,版本是稳定版的torch1.10。CUDA是10.2,最好严格按照10.2安装CUDA,11.3也一样。先复制下面的命令行备用。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值