Skip to content

Commit

Permalink
Update install_guide.md on pytorch3d
Browse files Browse the repository at this point in the history
Installing pytorch3d from conda worked but doing so from github/source failed for me (mapillary/inplace_abn#104).
  • Loading branch information
dgorissen authored Feb 12, 2024
1 parent e862efc commit 0e06d88
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion docs/prepare_env/install_guide.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 0e06d88

Please sign in to comment.