近期,安装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安装成功,输入下图命令,如下图的话就成功了
如何验证是否成功,我也是参考了众多的文章,然后自己尝试了一下。小白安装不易 ,看完后如果对你有帮助的话,就点个赞吧。