0,背景

Python3.10、cuda 11.7、pytorch 2.0.1

阅读【深度学习】【三维重建】windows10环境配置PyTorch3d详细教程-CSDN博客

1,解决方法

本来想尝试,结果发现CUB安装配置对照表里没有cuda 11.7对应的版本,不敢轻举妄动,所以没敢尝试,试了其他方法都没解决,最后在GIthub里找到了解决方案:Prebuilt wheels provided via 3rd party repository · facebookresearch/pytorch3d · Discussion #1752 · GitHub

指令:

pip install --extra-index-url https://miropsota.github.io/torch_packages_builder pytorch3d==0.7.6+pt2.0.1cu117

2,报错解决

刚开始报错:

以为是失败了,但一想这些的是无法连接服务器,就试了一下科学上网,然后就解决了 

3,测试是否成功

运行程序测试

import torch
from pytorch3d.structures import Meshes
from pytorch3d.utils import ico_sphere

# 创建一个简单的3D球体
sphere_mesh = ico_sphere(3, device=torch.device("cuda"))

# 打印球体的顶点和面信息
print("顶点坐标:", sphere_mesh.verts_packed())
print("面索引:", sphere_mesh.faces_packed())

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐