mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-02 03:42:50 +08:00
8b6310359f
Summary: Visualize a mesh in a plotly figure. - customize lighting and light position - customizable axis arguments (x, y, z) - customizable height and width of plotly figure - render batches of meshes in subplots or in a singular plot Reviewed By: nikhilaravi Differential Revision: D22611960 fbshipit-source-id: 5dc5c55e599d5b0d9c38f22e156c662654099e11
145 lines
4.2 KiB
Markdown
145 lines
4.2 KiB
Markdown
# Installation
|
||
|
||
|
||
## Requirements
|
||
|
||
### Core library
|
||
|
||
The core library is written in PyTorch. Several components have underlying implementation in CUDA for improved performance. A subset of these components have CPU implementations in C++/Pytorch. It is advised to use PyTorch3D with GPU support in order to use all the features.
|
||
|
||
- Linux or macOS or Windows
|
||
- Python 3.6, 3.7 or 3.8
|
||
- PyTorch 1.4, 1.5.0, 1.5.1 or 1.6.0
|
||
- torchvision that matches the PyTorch installation. You can install them together as explained at pytorch.org to make sure of this.
|
||
- gcc & g++ ≥ 4.9
|
||
- [fvcore](https://github.com/facebookresearch/fvcore)
|
||
- If CUDA is to be used, use at least version 9.2.
|
||
|
||
These can be installed by running:
|
||
```
|
||
conda create -n pytorch3d python=3.8
|
||
conda activate pytorch3d
|
||
conda install -c pytorch pytorch=1.6.0 torchvision cudatoolkit=10.2
|
||
conda install -c conda-forge -c fvcore fvcore
|
||
```
|
||
|
||
### Tests/Linting and Demos
|
||
|
||
For developing on top of PyTorch3D or contributing, you will need to run the linter and tests. If you want to run any of the notebook tutorials as `docs/tutorials` you will also need matplotlib.
|
||
- scikit-image
|
||
- black
|
||
- isort
|
||
- flake8
|
||
- matplotlib
|
||
- tdqm
|
||
- jupyter
|
||
- imageio
|
||
- plotly
|
||
|
||
These can be installed by running:
|
||
```
|
||
# Demos
|
||
conda install jupyter
|
||
pip install scikit-image matplotlib imageio plotly
|
||
|
||
# Tests/Linting
|
||
pip install black 'isort<5' flake8 flake8-bugbear flake8-comprehensions
|
||
```
|
||
|
||
## Installing prebuilt binaries for PyTorch3D
|
||
After installing the above dependencies, run one of the following commands:
|
||
|
||
### 1. Install with CUDA support from Anaconda Cloud, on Linux only
|
||
|
||
```
|
||
# Anaconda Cloud
|
||
conda install pytorch3d -c pytorch3d
|
||
```
|
||
|
||
Or, to install a nightly (non-official, alpha) build:
|
||
```
|
||
# Anaconda Cloud
|
||
conda install pytorch3d -c pytorch3d-nightly
|
||
```
|
||
### 2. Install from PyPI, on Linux and Mac
|
||
This works with pytorch 1.6.0 only.
|
||
```
|
||
pip install pytorch3d
|
||
```
|
||
On Linux this has support for CUDA 10.1. On Mac this is CPU-only.
|
||
|
||
## Building / installing from source.
|
||
CUDA support will be included if CUDA is available in pytorch or if the environment variable
|
||
`FORCE_CUDA` is set to `1`.
|
||
|
||
### 1. Install from GitHub
|
||
```
|
||
pip install 'git+https://github.com/facebookresearch/pytorch3d.git'
|
||
```
|
||
To install using the code of the released version instead of from the main branch, use the following instead.
|
||
```
|
||
pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'
|
||
```
|
||
|
||
**Install from Github on macOS:**
|
||
Some environment variables should be provided, like this.
|
||
```
|
||
MACOSX_DEPLOYMENT_TARGET=10.14 CC=clang CXX=clang++ pip install 'git+https://github.com/facebookresearch/pytorch3d.git'
|
||
```
|
||
|
||
### 2. Install from a local clone
|
||
```
|
||
git clone https://github.com/facebookresearch/pytorch3d.git
|
||
cd pytorch3d && pip install -e .
|
||
```
|
||
To rebuild after installing from a local clone run, `rm -rf build/ **/*.so` then `pip install -e .`. You often need to rebuild pytorch3d after reinstalling PyTorch.
|
||
|
||
**Install from local clone on macOS:**
|
||
```
|
||
MACOSX_DEPLOYMENT_TARGET=10.14 CC=clang CXX=clang++ pip install -e .
|
||
```
|
||
|
||
**Install from local clone on Windows:**
|
||
|
||
If you are using pre-compiled pytorch 1.4 and torchvision 0.5, you should make the following changes to the pytorch source code to successfully compile with Visual Studio 2019 (MSVC 19.16.27034) and CUDA 10.1.
|
||
|
||
Change python/Lib/site-packages/torch/include/csrc/jit/script/module.h
|
||
|
||
L466, 476, 493, 506, 536
|
||
```
|
||
-static constexpr *
|
||
+static const *
|
||
```
|
||
Change python/Lib/site-packages/torch/include/csrc/jit/argument_spec.h
|
||
|
||
L190
|
||
```
|
||
-static constexpr size_t DEPTH_LIMIT = 128;
|
||
+static const size_t DEPTH_LIMIT = 128;
|
||
```
|
||
|
||
Change python/Lib/site-packages/torch/include/pybind11/cast.h
|
||
|
||
L1449
|
||
```
|
||
-explicit operator type&() { return *(this->value); }
|
||
+explicit operator type& () { return *((type*)(this->value)); }
|
||
```
|
||
|
||
After patching, you can go to "x64 Native Tools Command Prompt for VS 2019" to compile and install
|
||
```
|
||
cd pytorch3d
|
||
python3 setup.py install
|
||
```
|
||
After installing, verify whether all unit tests have passed
|
||
```
|
||
cd tests
|
||
python3 -m unittest discover -p *.py
|
||
```
|
||
|
||
# FAQ
|
||
|
||
### Can I use Docker?
|
||
|
||
We don't provide a docker file but see [#113](https://github.com/facebookresearch/pytorch3d/issues/113) for a docker file shared by a user (NOTE: this has not been tested by the PyTorch3D team).
|