如何在cuda11.7上安装pytorch3d

创建虚拟环境

conda create -n pytorch3d python=3.9
conda activate pytorch3d

安装对应版本的torch

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple \
    https://download.pytorch.org/whl/cu116/torch-1.13.1%2Bcu116-cp39-cp39-linux_x86_64.whl \
    https://download.pytorch.org/whl/cu116/torchvision-0.14.1%2Bcu116-cp39-cp39-linux_x86_64.whl

安装pytorch3d

conda install https://anaconda.org/pytorch3d/pytorch3d/0.7.4/download/linux-64/pytorch3d-0.7.4-py39_cu116_pyt1131.tar.bz2

安装对应版本的numpy

pip install numpy==1.23.5