ubuntu系统下,libtorch环境加载.pt模型失败,一直报c10::Error错误

一、问题概括
ubuntu系统下,libtorch环境使用torch::jit::load(“xxx.pt”)接口加载模型失败,一直报c10::Error错误。

    std::cout<<"cudu support:"<< (torch::cuda::is_available()?"ture":"false")<<std::endl;
   try {
        torch::jit::script::Module module = torch::jit::load("xxx.pt");
	    module.to(torch::kCUDA);
	    module.eval();
    }
    catch (const c10::Error& e) {
        std::cerr << "Error loading the model11111!\n";
        std::exit(EXIT_FAILURE);
    }

二、问题解决
1、问题排查
1)怀疑是libtorch环境问题:
使用libtorch环境加载生成的其他模型,例如vgg、resnet18,均通过。libtorch环境“好像”没有问题。(后来发现此处有坑,虽然调用成功,但是可能是调用的cpu)
2)怀疑是模型转换有问题:
每一步骤挨着去过,没有发现问题。
3)再次怀疑libtorch环境问题:
无意在libtorch工程里打印了一下torch::cuda::is_available(),返回结果竟然是false! 终于发现问题,我的模型使用了gpu,但是现在libtorch环境下,gpu加载失败。
然后网上查找资料,最终定问问题是在cmake版本较低,导致libtorch无法调用gpu。
2、解决方法
1)升级cmake版本3.0.0以上
下载解压cmake压缩包,然后再建立软链接覆盖系统原来的cmake即可

# 下载cmake源码包
wget https://github.com/Kitware/CMake/releases/download/v3.21.4/cmake-3.21.4-linux-x86_64.tar.gz
tar -xzvf cmake-3.21.4-Linux-x86_64.tar.gz
# 将解压出来的包移到 /opt 目录下
sudo mv cmake-3.21.4-Linux-x86_64 /opt/cmake-3.21.4  
# 建立软链接
sudo ln -sf /opt/cmake-3.21.4/bin/* /usr/bin/   
cmake --version

cmake安装成功之后,更新 .bashrc:

# 进入~/.bashrc
sudo gedit ~/.bashrc
# 输入以下内容
export  PATH=$PATH:/opt/cmake-3.21.4/bin
#保存 .bashrc的更改和更新source一下
source ~/.bashrc 

2)cmakeList.txt里明确使用相应版本

#cmake_minimum_required(VERSION 2.6)  #修改前
cmake_minimum_required(VERSION 3.0 FATAL_ERROR)  #修改后

3)编译工程
重新对自己的工程cmke.,make。

又报出新错误NOTFOUND CUDA_cublas_device_LIBRARY,网上搜索,发现还是cmake版本问题,需要把cmake版本更换成 v3.14.3。因此重复步骤1重新安装cmake,然后步骤3),编译通过!!

  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
出现urllib.error.HTTPError: HTTP Error 403: Forbidden的错误提示,通常是由于访问被拒绝所导致的。这可能是由于服务器拒绝了你的请求,或者你没有足够的权限来访问该资源。 针对这个问题,可以尝试以下几种解决方法: 1. 检查URL是否正确:确保你正在访问正确的URL,并且该URL确实存在。 2. 检查网络连接:确保你的网络连接正常,并且你可以访问其他网站。 3. 检查权限:如果你正在尝试访问需要权限的资源,请确保你具有足够的权限来访问该资源。 4. 检查代理设置:如果你正在使用代理服务器,请确保代理服务器的设置正确,并且你可以访问该代理服务器。 5. 检查防火墙设置:如果你正在使用防火墙,请确保防火墙的设置正确,并且允许你访问该资源。 针对引用中的问题,可以尝试以下解决方法: 1. 检查PPA源是否正确:确保你添加的PPA源是正确的,并且你可以访问该源。 2. 检查网络连接:确保你的网络连接正常,并且你可以访问其他网站。 3. 检查权限:如果你正在尝试访问需要权限的资源,请确保你具有足够的权限来访问该资源。 4. 检查代理设置:如果你正在使用代理服务器,请确保代理服务器的设置正确,并且你可以访问该代理服务器。 5. 检查防火墙设置:如果你正在使用防火墙,请确保防火墙的设置正确,并且允许你访问该资源。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值