ubuntu 16.04 + CUDA11.3安装pytorch+pytorch3d

Kurt Sun2023-07-24 23:59

1. 建立环境

复制代码

conda create -n xxx python=3.8

因为是cuda11.3，所以python>=3.8。

复制代码

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

这里参照官方给出的方法。

复制代码

pip install fvcore iopath

复制代码

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

这里依旧是参考官方的方法。