预言(防踩坑)
恭喜你,比较幸运,安装GPU版本的PyTorch找到了此博文,博主最初因操作不利,由于自身轻微强迫症 + 完美主义强行刷机重装系统两次。
再次声明,用于NLP中的GPU加速,不用单独安装VS、CUDA、cuDNN。
请耐心看此博文,两步搞定安装!
第一步 准备一个python虚拟环境
若没有创建python虚拟环境请参考此博文
打开cmd输入activate python36
第二步 安装GPU版本的PyTorch
PyTorch官网
根据自己情况选择,CUDA千万不要选None,选了就变成CPU版本的PyTorch了
在cmd的python虚拟环境下输入上图标红即可!
耐心等待即可!
此博文只做学习用途,无任何商业用途!
大大怪爱小乖乖!
研哥哥