小白零基础入门版安装pytorch详解(非镜像快速安装)

        作为一个小白,安装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安装就可以,要切到自己的python版本环境中,否则就会出错。小白切记哦!

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

文本复制:

1、import torch
2、torch.cuda.is_available()
输出——True
3、 device = torch.device("cuda:0" if(torch.cuda.is_available() and True > 0) else "cpu")
4、print(device)
输出——cuda:0
5、 print(torch.cuda.get_device_name(0))
输出——GeForce GTX 1050 Ti

转载:https://blog.csdn.net/wobushizhainanyu/article/details/107778206

码字不易,希望大家多多点赞关注,后期会持续更新内容,谢谢大家。

猜你喜欢

转载自blog.csdn.net/hacker_NO_007/article/details/107780699