pytorch3d报错:RuntimeError: Not compiled with GPU support.

目录

解决方法:编译之前:加上指令:

[解决方法:pytorch3d 安装命令(ubuntu),成功!!!](#解决方法:pytorch3d 安装命令(ubuntu),成功!!!)

测试代码:


FORCE_CUDA=1 works for me. Thanks!

python 复制代码
      args = (
            points,
            cloud_to_packed_first_idx,
            num_points_per_cloud,
            image_size,
            radius,
            points_per_pixel,
            bin_size,
            max_points_per_bin,
        )
        # pyre-fixme[16]: Module `pytorch3d` has no attribute `_C`.
        idx, zbuf, dists = _C.rasterize_points(*args)
        ctx.save_for_backward(points, idx)
        ctx.mark_non_differentiable(idx)
        return idx, zbuf, dists

解决方法:编译之前:加上指令:

set FORCE_CUDA=1

报错:没有指定CUDA_HOME

File "C:\Users\002\.conda\envs\glige\lib\site-packages\torch\utils\cpp_extension.py", line 1048, in CUDAExtension library_dirs += library_paths(cuda=True) File "C:\Users\002\.conda\envs\glige\lib\site-packages\torch\utils\cpp_extension.py", line 1186, in library_paths paths.append(_join_cuda_home(lib_dir)) File "C:\Users\002\.conda\envs\glige\lib\site-packages\torch\utils\cpp_extension.py", line 2223, in _join_cuda_home raise EnvironmentError('CUDA_HOME environment variable is not set. ' OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.

方法1 重新安装命令 没测:

pip install "git+https://github.com/facebookresearch/pytorch3d.git"

方法2:

解决方法:pytorch3d 安装命令(ubuntu),成功!!!

pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py38_cu113_pyt1110/download.html

File "C:\Users\002\.conda\envs\glige\lib\site-packages\torch\autograd\function.py", line 539, in apply

return super().apply(*args, **kwargs) # type: ignore[misc]

File "C:\Users\002\.conda\envs\glige\lib\site-packages\pytorch3d-0.7.7-py3.8-win-amd64.egg\pytorch3d\renderer\points\rasterize_points.py", line 214, in forward

idx, zbuf, dists = _C.rasterize_points(*args)

RuntimeError: Not compiled with GPU support

测试代码:

python 复制代码
import torch
from pytorch3d import _C
from pytorch3d.structures import Pointclouds

import torch

# 修改后的参数准备
points_tensor = torch.tensor([[0.5000, 0.5000, 0.5000], [1.0000, 1.0000, 1.0000]])
other_tensor = torch.tensor([[1., 2., 3.], [4., 5., 6.]])
value_tensor = torch.tensor(0.1000)

another_tensor = torch.tensor([0., 0., 0.])
int_value1 = 10
int_value2 = 0
int_value3 = 0

# 调用函数
idx, zbuf, dists = _C.rasterize_points(
    points_tensor,
    other_tensor,
    value_tensor,
    [10, 10],
    another_tensor,
    int_value1,
    int_value2,
    int_value3)

Traceback (most recent call last):

File "E:\project\inpaint\Inpaint-Anything-main\aaa2.py", line 18, in <module>

idx, zbuf, dists = _C.rasterize_points(

^^^^^^^^^^^^^^^^^^^^

RuntimeError: a Tensor with 3 elements cannot be converted to Scalar

相关推荐
昀贝7 分钟前
Maven项目引用本地jar涉及scope和systemPath配置
python·maven·jar
Stuomasi_xiaoxin18 分钟前
服务器重装后如何“复活”旧硬盘上的 Anaconda 环境?—— 一次完整的排错与恢复记录
开发语言·python·github
Deng94520131418 分钟前
基于数据挖掘的课程推荐系统研究
人工智能·数据挖掘·数据预处理·基于用户的协同过滤·文本特征提取
这里有鱼汤23 分钟前
一招横盘突破选股法,赚钱不靠运气靠图形,靠概率!
后端·python
zhangfeng113329 分钟前
机器学习 YOLOv5手绘电路图识别 手绘电路图自动转换为仿真软件(如LT Spice)可用的原理图,避免人工重绘
人工智能·yolo·机器学习
0wioiw043 分钟前
Ubuntu基础(Python虚拟环境和Vue)
linux·python·ubuntu
xiao5kou4chang6kai41 小时前
Python-GEE遥感云大数据分析与可视化(如何建立基于云计算的森林监测预警系统)
python·数据分析·云计算·森林监测·森林管理
铭keny1 小时前
YOLO11 目标检测从安装到实战
人工智能·目标检测·目标跟踪
presenttttt1 小时前
用Python和OpenCV从零搭建一个完整的双目视觉系统(四)
开发语言·python·opencv·计算机视觉
木头左3 小时前
逻辑回归的Python实现与优化
python·算法·逻辑回归