Comment installer Pytorch3d sous Windows
Mon ordinateur portable est équipé d'un GPU Windows et n'a pas réussi à installer Pytorch3d. Il a fallu beaucoup d'efforts pour l'installer. Pour commémorer ma persévérance, je le partage avec tout le monde.
Installation de Pytorch3d sous WIN
1. Question :
Récemment, lorsque j'ai commencé la reconstruction 3D, j'ai découvert que de nombreux programmes de projets nécessitent la bibliothèque Pytorch3d. Son site officiel la définit comme suit :
PyTorch3D fournit des composants efficaces et réutilisables pour la recherche en vision par ordinateur 3D avec PyTorch.
On peut voir que la bibliothèque Pytorch3d peut fournir une variété de composants pour le traitement visuel 3D. Son installation sous Ubuntu est relativement simple, mais elle est un peu lourde sous Win. Cet article est un enregistrement de méthodes d'installation personnelles.
Contenu recommandé
2. Méthode d'installation
Points d'installation : correspondance stricte entre les versions cuda, pytorch et NVIDIA CUB
Liens requis pour l'installation :
- Versions de Pytorch3d : téléchargement des versions de Pytorch3d
- Téléchargement NVIDIA CUB : Téléchargement NVIDIA/cub
- Dans le lien de la version historique du site officiel de PyTorch mentionné ci-dessus ( Versions précédentes de PyTorch | PyTorch ), recherchez la commande d'installation de la v1.11.0. La commande d'installation de PyTorch correspondant à cuda11.3 est sélectionnée ici.
Avant l'installation, vous devez d'abord déterminer la correspondance des versions. Ma version installée est :
Boîte à outils CUDA 11.3 + pytorch 1.11.0 + Visual Studio 2019 + Pytorch3d 0.6.2 + Python 3.10
1. Pour NVIDIA CUB, la relation de version entre celui-ci et CUDA Toolkit est la suivante :
La version de cuda que j'ai installée est la 11.3, CUB a donc sélectionné la version correspondante 1.11.0. Après téléchargement et décompression, ajoutez son chemin à la variable d'environnement :
2. Installez les dépendances :
conda install -c fvcore -c iopath -c conda-forge fvcore iopath
3. Sélectionnez la version de Pytorch3d. Chaque version de Pytorch3d a sa version de pytorch applicable indiquée, comme indiqué ci-dessous :
L'image ci-dessus montre que les versions Pytorch applicables de pytorch3d 0.6.2 sont 1.10.1, 1.10.2, 1.11.0 et que le Python applicable est 3.10, mais que Pytorch1.6.0 et Python3.6 ne sont plus pris en charge. torch suivez toujours ceci Téléchargez la version pytorch3d correspondante
4. Après avoir installé VS 2019, ouvrez le terminal « Invite de commandes x64 Native Tools pour VS 2019 » comme indiqué ci-dessous, puis accédez au chemin du répertoire où pytorch3d est décompressé :
Entrez dans l'ordre :
Définissez les variables d'environnement dans l'invite de commande sous Windows :
définir DISTUTILS_USE_SDK=1
définir PYTORCH3D_NO_NINJA=1
Définissez les variables d'environnement dans PowerShell sous Windows :
$env:DISTUTILS_USE_SDK = 1Définissez les variables d'environnement sous Linux et macOS :
export DISTUTILS_USE_SDK=1
Ensuite, modifiez le code source du fichier setup.py dans pytorch3d : remplacez extra_compile_args = {"cxx": ["-std=c++14"]} par : extra_compile_args = {"cxx": []} ;
5. Entrez enfin dans le terminal « Invite de commandes x64 Native Tools pour VS 2019 » :
python setup.py installer --user
Le processus d'installation a pris environ 5 minutes sur mon ordinateur et la page n'arrêtait pas de sauter. Finalement, elle a finalement réussi :
Si votre installation échoue toujours, veuillez me contacter