Win11 Python3.10 安装pytorch3d
·
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())

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


所有评论(0)