Reputation: 21
I saw that more people seem to have the same issue, but it was not resolved. I am trying to install Pytorch3D with Anaconda and got the following PackageNotFound error.
Collecting package metadata (current_repodata.json): done Solving environment: failed with initial frozen solve. Retrying with flexible solve. Collecting package metadata (repodata.json): done Solving environment: failed with initial frozen solve. Retrying with flexible solve.
PackagesNotFoundError: The following packages are not available from current channels:
- pytorch3d
Current channels:
- https://conda.anaconda.org/pytorch3d/win-64
- https://conda.anaconda.org/pytorch3d/noarch
- https://repo.anaconda.com/pkgs/main/win-64
- https://repo.anaconda.com/pkgs/main/noarch
- https://repo.anaconda.com/pkgs/r/win-64
- https://repo.anaconda.com/pkgs/r/noarch
- https://repo.anaconda.com/pkgs/msys2/win-64
- https://repo.anaconda.com/pkgs/msys2/noarch
To search for alternate channels that may provide the conda package you're looking for, navigate to
https://anaconda.org
and use the search bar at the top of the page.
I have also tried using pip install 'git+https://github.com/facebookresearch/pytorch3d.git' and get the following:
C:\Users\Alexandra>pip install 'git+https://github.com/facebookresearch/pytorch3d.git'
ERROR: Invalid requirement: "'git+https://github.com/facebookresearch/pytorch3d.git'"
C:\Users\Alexandra>pip install git+https://github.com/facebookresearch/pytorch3d.git
Collecting git+https://github.com/facebookresearch/pytorch3d.git
Cloning https://github.com/facebookresearch/pytorch3d.git to c:\users\alexan~1\appdata\local\temp\pip-req-build-uspo7an4
Running command git clone -q https://github.com/facebookresearch/pytorch3d.git 'C:\Users\ALEXAN~1\AppData\Local\Temp\pip-req-build-uspo7an4'
ERROR: Error [WinError 2] The system cannot find the file specified while executing command git clone -q https://github.com/facebookresearch/pytorch3d.git 'C:\Users\ALEXAN~1\AppData\Local\Temp\pip-req-build-uspo7an4'
ERROR: Cannot find command 'git' - do you have 'git' installed and in your PATH?
I am on Windows 10, using python 3.8, PyTorch 1.5 and CUDA 10.2. I am very new to python, so I have no idea how to fix this (you can tell, that I've never installed from git before..) (please be lenient!) Thank you!
EDIT: Thank you for your answers. I did install Git and it got me a bit further, but still not completeing the build..
Also, interesting enough, when I run the commands
!pip install torch torchvision
!pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'
in Google Collab it seems to work, but I cannot run it , let's say, in jupyter. Any more ideas?
Upvotes: 1
Views: 18232
Reputation: 76
I was just now able to get the pytorch3d installation in windows 11 via PIP. Here are the version details
torch 1.13.1+cu116
torchvision 0.14.1+cu116
fvcore 0.1.5.post20221221
iopath 0.1.10
pytorch3d 0.7.2
NOTE: fvcore and iopath will be installed directly while installing pytorch3d (step4)
Installation steps:
install pytorch 1.13.1 with cuda 11.6 (make sure the gpu driver is uptodate)
python -m pip install torch==1.13.1 torchvision --index-url https://download.pytorch.org/whl/cu116
install the cuda toolkit version 11.6 fro windows 11 from official source
set the CUDA_HOME path
# if powershell
$env:CUDA_HOME="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6"
# if windows cmd terminal
set CUDA_HOME=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6
git clone --branch v0.7.2 https://github.com/facebookresearch/pytorch3d.git
cd pytorch3d
git cherry-pick 3388d3
python setup.py install
Upvotes: 0
Reputation: 699
Recently, I succeded installing pytorch3d in windows 10/11 with latest version of pytorch(1.13.1).
As Pytorch3D has some dependency with cuda, it has a little complexity to build from source. But it can be solved anyway.
(1) First, install cuda 11.6 windows and check enviroment variable.
As someone pointed in this article, CUB_HOME path variable is not needed, as we use 11.6 cuda toolkit.
(2) Then install pytorch latest, in my case 1.13.1 with conda tool.(When I tried pip version, it was not successful.)
conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia
(3) Install needed packages with Conda. You can check it with INSTALL.md in pytorh3d source.
conda install -c fvcore -c iopath -c conda-forge fvcore iopath
(4) Check build tool. we can build it with Visual Studio 2019 C++ package with compiler v142. you can check it with Visual Studio Installer.
(5) Run x64 native build tool then move to pytorch3d source
(6) install it with ...
python setup.py install
Upvotes: 2
Reputation: 4268
With CUDA 11.6 downloading CUB
and setting CUB_HOME
is no longer necessary. Trying to use CUB_HOME
will give nvcc.exe
compile error. Any previous CUB_HOME
environment variable should be deleted and command line restarted before running setup.
I have also tried to install pytorch3d on windows 10. As of writing this there is no windows package in, https://anaconda.org/pytorch3d/pytorch3d.
Pytorch3d install doc has detailed instructions but some information is missing and only found inside various issues. Following various issues I was able get pytorch3d installed by compiling from source on pytorch 1.8.1
and 1.10.0
(This version is not supported yet in official docs for pytorch3d 0.6.0
).
I have tested on pytorch 1.8.1 with CUDA 10.2 and pytorch 1.10.0 with CUDA 11.3. I had CUDA Toolkit 11.0, CuDNN installed separately with environment variables set to be used by tensorflow gpu. For both environment a new python 3.9 was used. Visual studio 16.11.5 was used with Desktop Development with C++
enabled, CMake 3.21.3.
It is probably better to have same CUDA Toolkit version as the Pytorch GPU version. There was warning regarding version but in my case it was installed successfully.
The pytorch3d source code must be downloaded and extracted in order to compile. When running python setup.py install
from pytorch3d folder it looked for CUDA_HOME. It was able to find correct CUDA path probably based on other flags.
I initially faced these errors,
conda create -n pytorch3d python=3.9
conda activate pytorch3d
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
conda install -c fvcore -c iopath -c conda-forge fvcore iopath
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\extras\CUPTI\lib64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\include
cl.exe
Visual C++ compiler. This will allow to run compilation from conda prompt with correct environment selected. Host x64 was used and x86 for the folder inside as x64 one gave error for me. In my case it was,C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\MSBuild\Current\Bin
C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Tools\MSVC\14.29.30133\bin\Hostx64\x86
Following this issue, PYTORCH3D_NO_NINJA
system environment variable was set with value of 1
.
FORCE_CUDA
with value of 1
(though not required if CUDA is available in pytorch) and CUB_HOME
system environment variable was set by downloading it from, https://github.com/NVIDIA/cub/releases.
PYTORCH3D_NO_NINJA 1
CUB_HOME C:\portable\cub-1.9.9
CUDA_PATH_V11_0 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0
CUDA_PATH C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0
Following this issue instead of removing -std=c++14
from extra_compile_args
in setup.py
, commenting out "-std=c++14"
in nvcc_args
worked for me. I did check if the suggested method in the issue works.
Now running python setup.py install
from pytorch3d source folder should start compile and install it.
Install following requirements in conda env to run demos and examples.
conda install jupyter
pip install scikit-image matplotlib imageio plotly opencv-python
Some of the steps mentioned above is probably not needed. There is no need to modify any header files. This is the source that worked for me, https://github.com/facebookresearch/pytorch3d/tree/bfeb82efa38f29ed5b9cf8d8986fab744fe559ea.
Upvotes: 5
Reputation: 1
https://github.com/facebookresearch/pytorch3d/blob/master/INSTALL.md
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
Upvotes: 0