PyTorch 1.7 CUDA 11编译PyTorch-EMDLoss

最新推荐文章于 2024-06-19 09:35:21 发布
最新推荐文章于 2024-06-19 09:35:21 发布
阅读量2.7k 收藏 4
点赞数 1
文章标签： pytorch 深度学习 python
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weijiaw/article/details/122998806
版权

尝试编译并安装PyTorch的C++扩展：EMD Loss （Earth Mover’s Distance 动土距离），GitHub链接：https://github.com/meder411/PyTorch-EMDLoss

本人使用Ubuntu 16.04，PyTorch 1.7 + CUDA 11，但由于原文件是用PyTorch 1.0写的，所以直接运行python setup.py install的话，ninja编译器会报错。报错很长，这里贴几个关键词：

（中间）

FAILED: /xxxxx/PyTorch-EMDLoss-master/build/temp.linux-x86_64-3.6/pkg/src/emd.o

（结尾处）

File "/xxxxx/anaconda3/envs/pcrnet/lib/python3.6/subprocess.py", line 438, in run
output=stdout, stderr
最低0.47元/天 解锁文章
王小熊嗷
关注
  • 1
    点赞
  • 踩
  • 4
    收藏
    觉得还不错? 一键收藏
  • 6
    评论
EMD loss
hzhj的博客
05-15 6395
Earth Mover's Distance (EMD)，翻译过来是地球移动距离，又称为推土机距离，是对特征空间中两个多维矩阵中某一维距离的一种度量。 The Earth Mover's Distance (EMD) is a method to evaluate dissimilarity between two multi-dimensional distributions in some feature space where a distance measure between single f
EMD的matlab代码分享-PyTorchEMD:用于3D点云回归的Earth-Mover-Distance(EMD)的PyTorchWra
06-12
EMD的matlab代码分享用于点云地球移动距离 (EMD) 的 PyTorch 包装器 依赖 代码已经在 Ubuntu 16.04、PyTorch 1.1.0、CUDA 9.0 上测试过。 用法 首先编译使用 python setup.py install 然后，将lib文件复制到主目录， cp build/lib.linux-x86_64-3.6/emd_cuda.cpython-36m-x86_64-linux-gnu.so . 然后，您可以简单地使用它 from emd import earth_mover_distance d = earth_mover_distance(p1, p2, transpose=False) # p1: B x N1 x 3, p2: B x N2 x 3 例如检查test_emd_loss.py 。 作者 cuda代码最初由范浩强编写。 PyTorch 包装器是由莫开春编写的。此外，顾家远也提供了帮助。 执照 麻省理工学院
探索深度学习中的距离新维度：PyTorch EMDLoss 框架
最新发布
gitblog_00043的博客
06-19 703
探索深度学习中的距离新维度：PyTorch EMDLoss 框架 项目地址:https://gitcode.com/meder411/PyTorch-EMDLoss 项目介绍 在深度学习的广阔天地里，衡量数据点间相似性的度量是模型训练和评估的关键。今天，我们聚焦于一款名为 PyTorch EMDLoss 的开源项目，它为Python社区带来了高效计算近似地球移动距离（Earth Mover's D...
Pytorch中torch.nn的损失函数
12-21
目录 前言 一、torch.nn.BCELoss(weight=None, size_average=True) 二、nn.BCEWithLogitsLoss(weight=None, size_average=True) 三、torch.nn.MultiLabelSoftMarginLoss(weight=None, size_average=True) 四、总结 前言 最近使用Pytorch做多标签分类任务，遇到了一些损失函数的问题，因为经常会忘记（好记性不如烂笔头囧rz），都是现学现用，所以自己写了一些代码探究一下，并在此记录，如果以后还遇到其他损失函数，继续在此补充。 如果有兴趣，我建
pytorch搭建emd+tcn+lstm 实现时间序列预测 完整输入输出
mqdlff_python的博客
01-10 561
pytorch搭建emd+tcn+lstm 实现时间序列预测 完整输入输出
风速预测（二）基于Pytorch的EMD-LSTM模型
小蜗爱建模的博客
12-14 2142
本文基于前期介绍的风速数据（文末附数据集），先经过经验模态EMD分解,然后通过数据预处理，制作和加载数据集与标签，最后通过Pytorch实现EMD-LSTM模型对风速数据的预测。
Python库 | monk_pytorch_cuda90_test1-0.0.1.tar.gz
03-06
《Python库monk_pytorch_cuda90_test1-0.0.1的深度解析》 在当今快速发展的信息技术领域，Python已经成为数据科学、人工智能（AI）和机器学习（ML）领域的首选编程语言。其中，PyTorch作为一款强大的深度学习框架，...
pytorch - py3.9- cuda12.4 - linux
06-13
pytorch - py3.9- cuda12.4 - linux - pytorch - py3.9- cuda12.4 - linux - pytorch - py3.9- cuda12.4 - linux - pytorch - py3.9- cuda12.4 - linux - pytorch - py3.9- cuda12.4 - linux - pytorch - py3.9- ...
a-PyTorch-Tutorial-to-Image-Captioning-master_pytorch_
09-30
1. **动态计算图**：PyTorch的动态计算图允许在运行时构建和修改计算图，而numpy是静态的，计算图在编译时确定。 2. **自动求梯度**：PyTorch可以自动计算梯度，而numpy不提供此功能，需要手动实现反向传播。 3. *...
jetson nano pytorch 1.7 网盘链接
02-05
Python 3.6 - torch-1.7.0-cp36-cp36m-linux_aarch64.whl 适用于：JetPack 4.4 (L4T R32.4.3) / JetPack 4.4.1 (L4T R32.4.4) / JetPack 4.5 (L4T R32.5.0)
emd算法代码
03-26
emd全套matlab代码
Python库 | pytorch3d-0.6.1-cp39-cp39-macosx_10_9_x86_64.whl
02-19
它提供的whl文件是预编译的二进制包，解压后可以直接在Python环境中使用，大大简化了安装过程。 **3D深度学习与PyTorch3D** 3D深度学习是计算机视觉领域的一个重要分支，它处理三维数据，如点云、多视图几何和3D...
Python实现长短记忆神经网络（LSTM）预测经验模态分解（EMD）各模态变化的组合时间序列预测模型
weixin_40169609的博客
06-09 8601
本实验使用环境为Anaconda3 Jupyter，调用Sklearn包、Keras包，请提前准备好。只提供数据格式而不提供数据，本人是代码缝合怪小白，望大牛指点。 1.导包 主要包含pandas、numpy、绘图包、日期格式、数学计算、pyhht 、标准化、验证函数等包。有一些包可能没啥用…忘了什么时候导入的了。 import csv import numpy as np import time from sklearn.preprocessing import StandardScaler from s
pytorch——计算两个等大无序点云中的距离Earth Mover Distance
Irving.Gao的博客
10-02 1278
目前要做两个无序点集之间的相似性计算，在看过Chamfer Distance后，个人觉得CD的计算方式决定了其无法处理两个点集整体分布差异大，但是局部有部分点距离很近的情况，而这种情况在点集中是一定可能出现的，因此使用EMD可以保证每个点集都有一一对应的整体cost最小点。
ubuntu16.04安装cuda和pytorch时需要注意的坑
j_l_sheng的博客
03-17 1481
在ubuntu16.04上安装cuda和pytorch时，一不小心就会出现意想不到的问题，比如屏幕分辨率突然变的很小且不可调、在登录界面一直循环登录不了、pytorch调用cuda失败等问题。具体解决方案已经整理发布，这篇文章再来总结一下需要考虑的细节。 问题总结： 显卡与驱动的版本要匹配，切忌根据nvidia-smi命令所显示的信息来判断所需的cuda版本 得到图片中的信息后，我立马下载安装了cuda11.2，结果出现了屏幕分辨率降低、在登录界面循环的问题。 问题关键： cuda11.2与nvidi
CUDA加速的CD,EMD计算(pytotrch版本)
冯子材的博客
06-28 2653
我在我的github给出了cuda加速的CD与EMD计算代码.可以作为插件集成到自己的pytorch工程.
emd cuda 使用记录
qq_39964246的博客
08-23 551
emd cuda 记录 由于刚开始在服务器上使用的是cuda-8.0，make文件时，总会报错__shfl_down is undefined for cuda, 经查证后得出，warp shuffle 函数（__shfl、 shfl_up、__shfl_down和 __shfl_xor）和warp vote 函数（__any、__all、__ballot）已删除，不再适用于cuda8.0及更早的版本. Warp shuffle内在函数仅仅是定义（仅受支持）计算能力（cc）3.0架构及更高版本。在CUDA
点云损失函数Chamfer Distance 和 Earth Mover‘s Distance
qq_39964246的博客
08-12 5426
浅析点云损失函数Chamfer Distance 和 Earth Mover‘s Distance
mamba install pytorch torchvision pytorch-cuda=11.7 pytorch3d -c pytorch -c nvidia -c pytorch3d
07-28
根据引用\[1\]和引用\[2\]的内容，你可以使用以下命令在虚拟环境中安装PyTorch和相关库： ``` mamba install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia ``` 这个命令会安装PyTorch、torchvision和torchaudio，并指定使用CUDA 11.7版本。同时，它会从pytorch和nvidia的频道中获取软件包。 然而，根据引用\[3\]的内容，如果你在指定的镜像源中找不到指定版本的PyTorch，可能会导致安装的是CPU版本而不是GPU版本。为了解决这个问题，你可以尝试使用其他镜像源或者手动指定安装GPU版本的PyTorch。 综上所述，你可以尝试使用以下命令来安装PyTorch和相关库，并指定使用CUDA 11.7版本： ``` mamba install pytorch torchvision pytorch-cuda=11.7 -c pytorch -c nvidia ``` 希望这能帮到你！ #### 引用[.reference_title] - *1* [三分钟搞懂最简单的Pytorch安装流程](https://blog.csdn.net/weixin_44261300/article/details/129643480)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control,239^v3^insert_chatgpt"}} ] [.reference_item] - *2* [Pytorch与NVIDA驱动控制安装](https://blog.csdn.net/m0_48176714/article/details/129311194)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control,239^v3^insert_chatgpt"}} ] [.reference_item] - *3* [解决使用conda下载pytorch-gpu版本困难的问题](https://blog.csdn.net/qq_41963301/article/details/131070422)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]
评论 6
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值