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())
相关推荐
子燕若水1 小时前
Unreal Engine 5中的AI知识
人工智能
极限实验室2 小时前
Coco AI 实战(一):Coco Server Linux 平台部署
人工智能
杨过过儿2 小时前
【学习笔记】4.1 什么是 LLM
人工智能
巴伦是只猫2 小时前
【机器学习笔记Ⅰ】13 正则化代价函数
人工智能·笔记·机器学习
大千AI助手3 小时前
DTW模版匹配:弹性对齐的时间序列相似度度量算法
人工智能·算法·机器学习·数据挖掘·模版匹配·dtw模版匹配
AI生存日记3 小时前
百度文心大模型 4.5 系列全面开源 英特尔同步支持端侧部署
人工智能·百度·开源·open ai大模型
LCG元3 小时前
自动驾驶感知模块的多模态数据融合:时序同步与空间对齐的框架解析
人工智能·机器学习·自动驾驶
why技术3 小时前
Stack Overflow,轰然倒下!
前端·人工智能·后端
烛阴4 小时前
简单入门Python装饰器
前端·python