小白离线快速安装GPU版pytorch(不用镜像源,解决安装缓慢问题)

风格不统一 提交于 2020-08-12 08:53:03

近期,安装pytorch入了很多坑,清华的镜像源不能用,这就很烦人了。我尝试了很多种,中科大源也尝试了,但是 还是不行。后来呢,我就去尝试离线安装,最开始在清华的镜像网下载的离线安装包,用conda命令安装后出现了caffe2_detectron_ops_gpu.dll模块加载不了或者缺少依赖项。(这个问题我查了1整天,见到过几个这个问题,但是由于人家用的是Mac的系统,我就更难受了)最后,我去官网下载的对应版本的离线安装包,然后就安装成功了。(忒棒嘞)详细步骤如下:

1、在安装GPU版本pytorch之前先安装对应版本的CUDA和CUDNN

(详细安装见我的CUDA和CUDNN的安装文章https://blog.csdn.net/wobushizhainanyu/article/details/107736090

2、去pytorch官网找到对应要安装的版本https://pytorch.org/

复制对应版本的后边的那个链接,浏览器打开

3、打开上述链接后找到对应版本的包下载torch和torchvision版本需要对应,例如我的

4、然后打开命令行 cd切换目录至上述两个包所下载的位置 然后pip安装就可以

5、验证pytorch安装成功,输入下图命令,如下图的话就成功了

如何验证是否成功,我也是参考了众多的文章,然后自己尝试了一下。小白安装不易 ,看完后如果对你有帮助的话,就点个赞吧。

 

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!