小白零基础入门版安装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

 

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

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值