1. 建立环境
conda create -n xxx python=3.8
因为是cuda11.3,所以python>=3.8。
2. 安装pytorch等包
conda install cudatoolkit=11.3 matplotlib matplotlib-base
这里我选择把matplotlib等安装上,不需要的话请忽略。cudatoolkit是习惯上用conda安装,pytorch torchvision习惯用pip安装。
pip install torch==1.12.0+cu113 torchvision==0.13.0+cu113 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu11
这里参照官方给出的方法。
3. 安装pytorch3d的基础包
pip install fvcore iopath
4. 安装pytorch3d
pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py38_cu113_pyt1120/download.html
这里依旧是参考官方的方法。