Linux-Anaconda虚拟环境 安装配置GPU版本Pytorch 流程+遇到与解决问题小结
遇到问题最近在用pytorch跑深度学习模型,不过一直用的是CPU,训练速度还是跟不上,正好实验室服务器比较空闲,而且cuda、cudnn也都是配置好了的,所以打算在上面的anaconda虚拟环境中配置一下我想要用的gpu版本的pytorch,配置期间遇到了不少问题,也参考了很多博主的经验,现自己总结记录一下。总结先说结论,代码可以跑起来了就会发现,事情怎么这么简单(笑哭),但其实是将近一周的折腾才搞好的。因为我只想连服务器的gpu用一下,上面cuda、cudnn都已经配置好了的,我只需要装一个gpu
原创
2021-04-15 11:32:41 ·
3139 阅读 ·
0 评论