pytorch3d 安装报错 RuntimeError: Not compiled with GPU support pytorch3d

安装环境

NVIDIA GeForce RTX 3090
cuda 11.3
python 3.8.5
torch 1.11.0
torchvision 0.12.0

环境安装命令

 conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch

安装pytorch3d参考官网链接

https://github.com/facebookresearch/pytorch3d/blob/main/INSTALL.md

1. conda install pytorch3d -c pytorch3d 失败

2. 先克隆pytorch包再编译可以安装成功，但是函数运行报错 RuntimeError: Not compiled with GPU support pytorch3d

报错信息

pix_to_face, zbuf, barycentric_coords, dists = _C.rasterize_meshes( RuntimeError: Not compiled with GPU support pytorch3d

报错截图

解决方法：pytorch3d 安装命令（ubuntu），成功！！！

pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py38_cu113_pyt1110/download.html