diff --git a/INSTALL.md b/INSTALL.md index d9428223..5f34ae4d 100644 --- a/INSTALL.md +++ b/INSTALL.md @@ -8,8 +8,8 @@ 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, 1.6.0, 1.7.0, or 1.7.1. +- Python 3.6, 3.7, 3.8 or 3.9 +- PyTorch 1.4, 1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1 or 1.9.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) @@ -78,17 +78,18 @@ Or, to install a nightly (non-official, alpha) build: conda install pytorch3d -c pytorch3d-nightly ``` ### 2. Install from PyPI, on Mac only. -This works with pytorch 1.7.1 only. The build is CPU only. +This works with pytorch 1.9.0 only. The build is CPU only. ``` pip install pytorch3d ``` ### 3. Install wheels for Linux -We have prebuilt wheels with CUDA for Linux for PyTorch 1.7.0 and 1.7.1, for each of the CUDA versions that they support. +We have prebuilt wheels with CUDA for Linux for PyTorch 1.9.0, for each of the CUDA versions that they support, +for Python 3.7, 3.8 and 3.9. These are installed in a special way. -For example, to install for Python 3.6, PyTorch 1.7.0 and CUDA 10.1 +For example, to install for Python 3.8, PyTorch 1.9.0 and CUDA 10.2 ``` -pip install pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py36_cu101_pyt170/download.html +pip install pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py38_cu102_pyt190/download.html ``` In general, from inside IPython, or in Google Colab or a jupyter notebook, you can install with @@ -138,32 +139,9 @@ 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. +Depending on the version of PyTorch, changes to some PyTorch headers may be needed before compilation. These are often discussed in issues in this repository. -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 +After any necessary patching, you can go to "x64 Native Tools Command Prompt for VS 2019" to compile and install ``` cd pytorch3d python3 setup.py install