您现在的位置是：首页 >技术杂谈 >Pytorch3D指南~网站首页技术杂谈

0、Pytorch3D介绍

https://github.com/facebookresearch/pytorch3d

1、安装
参考链接：
https://blog.csdn.net/xiao_ZHEDA/article/details/144752827
https://www.bilibili.com/opus/903477294519025683
我是直接下载whl文件来安装的，
下载地址：https://miropsota.github.io/torch_packages_builder/pytorch3d/
下载之前查看自己的cuda版本，下载对应版本的文件

import torch
# 检查 PyTorch 是否支持 CUDA
print("PyTorch supports CUDA:", torch.cuda.is_available())

# 获取 CUDA 版本
print("CUDA version:", torch.version.cuda)

# 获取可用的 GPU 设备数量
print("Number of available GPUs:", torch.cuda.device_count())

# 获取每个 GPU 的名称和属性
for i in range(torch.cuda.device_count()):
    print(f"GPU {i}: {torch.cuda.get_device_name(i)}")

下载好了，直接pip install xxx.whl即可
安装成功之后可以看下版本，确认是否安装成功

import pytorch3d
print(pytorch3d.__version__)