Win11 Python3.10 安装pytorch3d

sagima_sdu2024-07-03 9:07

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())