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