Pytorch3D环境血泪安装史_pytorch3d conda-forge

Pytorch3D环境血泪安装史

conda create -n pytorch3d python=3.8
conda activate pytorch3d
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch
conda install -c conda-forge -c iopath fvcore iopath
conda install -c pytorch3d pytorch3d
1
2
3
4
5

版本全要对齐，尤其注意cudatoolkit=10.1，可以避免一系列奇怪的报错！！

另一种，方法，如果显卡是3090等一定要基于cudatoolkit11.0以上的，需要用wheels安装

conda create -n pytorch3d python=3.8
conda activate pytorch3d
conda install pytorch torchvision torchaudio -c pytorch
conda install -c conda-forge -c iopath fvcore iopath
pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py38_cu113_pyt1110/download.html
1
2
3
4
5