这篇文章主要为大家展示了“如何安装Pytorch”,内容简而易懂,条理清晰,希望能够帮助大家解决疑惑,下面让小编带领大家一起研究并学习一下“如何安装Pytorch”这篇文章吧。
首先打开Pytorch的官网:https://pytorch.org/。在首页我们可以看到有各种配置可选,我们这里选择CPU或是GPU的版本都可以。一般选择GPU版本的Pytorch,这样运行大型的程序速度会快很多。而要运行GPU的Pytorch,就需要在电脑(不管是Ubuntu系统还是windows系统)安装相应的CUDA9和cudnn7,这两个安装起来有时候会遇到一大堆问题,所以需要一步一步解决。推荐一个我写的在Ubuntu系统安装教程的博客,按照博客上写的一步步来就可以啦。博客链接:https://blog.csdn.net/xiewenrui1996/article/details/102736238。
其实还有一个更简洁的安装方法,就是安装anaconda,然后在anaconda创建虚拟环境,在虚拟环境里面安装GPU版的Pytorch,系统就会自动给你安装相应的CUDA和cudnn,这样就节省了一大半的时间。写程序的编译器推荐用Pycharm,里面集成了许多写程序需要的模块,简单实用。Pycharm可以跟在anaconda里面创建的GPU版的Pytorch相连起来,把创建的虚拟环境在Pycharm使用,具体操作为在编译器里面添加anaconda的路径就可以了。
以上是“如何安装Pytorch”这篇文章的所有内容,感谢各位的阅读!相信大家都有了一定的了解,希望分享的内容对大家有所帮助,如果还想学习更多知识,欢迎关注亿速云行业资讯频道!
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。