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

预言(防踩坑)

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

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

若没有创建python虚拟环境请参考此博文
打开cmd输入activate python36
Windows下安装GPU版本的PyTorch(防踩坑)_第1张图片

第二步 安装GPU版本的PyTorch

PyTorch官网
根据自己情况选择,CUDA千万不要选None,选了就变成CPU版本的PyTorch了
Windows下安装GPU版本的PyTorch(防踩坑)_第2张图片
在cmd的python虚拟环境下输入上图标红即可!
耐心等待即可!
此博文只做学习用途,无任何商业用途!
大大怪爱小乖乖!
研哥哥

你可能感兴趣的:(Python,GOU版PyTorch)