2021-10-12 服务器端配置深度、位姿联合无监督预测模型的代码_笔记(显卡、cuda、pytorch匹配_屏蔽warning)

主要问题:
算法中用到 Spatial correlation sampler 包,使用 pip 在 torch1.6.0环境下安装,不可用。在之前使用 pytorch1.3.1与编译安装包(Pytorch-Correlation-extension-master)的方法在TitanXp上成功运行。编译的方法依然可以通过。

解决方法:
在保证显卡驱动:最新Cuda:10.1Torch:1.6.0的情况下, 最终在服务器虚拟环境内安装GCC-5.4.0重新编译Pytorch-Correlation-extension-master,即可成功运行。

Tips:

  • 根据cuda版本查找对应的pytorch,https://pytorch.org/get-started/previous-versions/

  • Cuda与nvidia驱动匹配:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html,驱动是向下匹配,安装最新即可。

  • GCC版本不匹配,可以用Conda安装,再建立软链。

    • https://www.zhihu.com/question/56272908(安装最新GCC)
    • https://blog.csdn.net/j___t/article/details/107308883(安装GCC + G++:5.4.0)
  • torch.cuda.is_available()查看Cuda是否可用。

  • 显卡算力查询:https://developer.nvidia.com/zh-cn/cuda-gpus

  • 查看Cudnn版本:

     cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
    
  • 不显示warning

    import warnings
    warnings.filterwarnings("ignore")
    
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

BlueagleAI

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值