Windows下安装GPU版本的PyTorch(防踩坑)

预言(防踩坑)

恭喜你,比较幸运,安装GPU版本的PyTorch找到了此博文,博主最初因操作不利,由于自身轻微强迫症 + 完美主义强行刷机重装系统两次。
再次声明,用于NLP中的GPU加速,不用单独安装VS、CUDA、cuDNN。
请耐心看此博文,两步搞定安装!

第一步 准备一个python虚拟环境

若没有创建python虚拟环境请参考此博文
打开cmd输入activate python36
在这里插入图片描述

第二步 安装GPU版本的PyTorch

PyTorch官网
根据自己情况选择,CUDA千万不要选None,选了就变成CPU版本的PyTorch了
在这里插入图片描述
在cmd的python虚拟环境下输入上图标红即可!
耐心等待即可!
此博文只做学习用途,无任何商业用途!
大大怪爱小乖乖!
研哥哥

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值