在Win10安装pytorch3d

最新推荐文章于 2024-08-23 19:52:04 发布

hero poi

最新推荐文章于 2024-08-23 19:52:04 发布

阅读量922

点赞数 2

文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/qq_47183165/article/details/130012053

版权

Pytorch3d是一个为Pytorch设计的库，专注于3D视觉研究，提供三角形网格数据结构、操作、可区分的渲染器和隐式函数合成。它支持GPU加速和小批量异构数据处理，适用于深度学习中的3D数据预测和操作。

摘要由CSDN通过智能技术生成

Pytorch3d

这是一个为使用Pytorch的3D计算机视觉研究提供的高效可重用的组件

主要特点包括：

由于pytorch3d旨在用于预测和操作3D数据的深度学习方法的顺利集成，因此在其中的运算符：

以上是对于这个包的简要介绍，需要了解更多，可以点击官方链接

以下操作都是在anaconda虚拟环境中进行的，如果有小伙伴想要建立虚拟环境可以参考：链接

首先需要检查自己的cuda以及pytorch版本

cuda安装地址：https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local

如果已经装好，检查自己的torch和cuda版本的是否可用

import torch
torch.cuda.is_available()

如果返回是True则已经在使用GPU版本的torch

官网对比版本查询

如果返回不是True，则需要卸载重新安装

pip uninstall torch torchvision torchaudio
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

最后再用**torch.cuda.is_available()**检查即可

1.conda install -c fvcore -c iopath -c conda-forge fvcore iopath

2.进入下载地址，在最下面选择自己选哟的版本

在这里插入图片描述
3.再下拉到最下面进行下载

4.解压后在文件夹中

python setup.py install

在这里插入图片描述
此时import pytorch3d 也是ok的

在这里插入图片描述

关注