From 0e06d883679cbda1558e9882f86444c6423ba5fc Mon Sep 17 00:00:00 2001 From: Dirk Gorissen Date: Mon, 12 Feb 2024 19:16:25 +0000 Subject: [PATCH] Update install_guide.md on pytorch3d Installing pytorch3d from conda worked but doing so from github/source failed for me (https://github.com/mapillary/inplace_abn/issues/104). --- docs/prepare_env/install_guide.md | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/docs/prepare_env/install_guide.md b/docs/prepare_env/install_guide.md index 67f2df4..7427262 100644 --- a/docs/prepare_env/install_guide.md +++ b/docs/prepare_env/install_guide.md @@ -20,7 +20,10 @@ conda install conda-forge::ffmpeg # ffmpeg with libx264 codec to turn images to ### We recommend torch2.0.1+cuda11.7. conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia -# Build from source, it may take a long time (Proxy is recommended if encountering the time-out problem) +# Install from pytorch3d from conda +conda install pytorch3d::pytorch3d + +# Alternatively, build from source, it may take a long time (Proxy is recommended if encountering the time-out problem) pip install "git+https://github.com/facebookresearch/pytorch3d.git@stable" # MMCV for some network structure