Cómo instalar Pytorch3d en Windows
Mi computadora portátil tiene una GPU de Windows y no pudo instalar Pytorch3d. Me costó mucho instalarlo. Para conmemorar mi perseverancia, lo comparto con todos.
Instalación de Pytorch3d bajo WIN
1 pregunta:
Recientemente, cuando comencé la reconstrucción 3D, descubrí que muchos programas de proyectos requieren la biblioteca Pytorch3d, su sitio web oficial la define como:
PyTorch3D proporciona componentes eficientes y reutilizables para la investigación de visión por computadora 3D con PyTorch.
Se puede ver que la biblioteca Pytorch3d puede proporcionar una variedad de componentes para el procesamiento visual 3D. Su instalación en Ubuntu es relativamente simple, pero un poco engorrosa en Win. Esta publicación es un registro de los métodos de instalación personales.
Contenido recomendado
2. Método de instalación
Puntos de instalación: correspondencia estricta entre las versiones de cuda, pytorch y NVIDIA CUB
Enlaces necesarios para la instalación:
- Versiones de Pytorch3d: Descarga de versiones de Pytorch3d
- Descarga de NVIDIA CUB: Descarga de NVIDIA/cub
- En el enlace de la versión histórica del sitio web oficial de PyTorch mencionado anteriormente ( Versiones anteriores de PyTorch | PyTorch ), busque el comando de instalación de v1.11.0. Aquí se selecciona el comando de instalación de PyTorch correspondiente a cuda11.3.
Antes de la instalación, primero debe determinar la correspondencia de la versión. Mi versión instalada es:
Kit de herramientas CUDA 11.3 + pytorch 1.11.0 + Visual Studio 2019 + Pytorch3d 0.6.2 + Python 3.10
1. Para NVIDIA CUB, la relación de versión entre este y CUDA Toolkit es la siguiente:
La versión de cuda que instalé es 11.3, por lo que CUB seleccionó la versión 1.11.0 correspondiente. Después de descargarla y descomprimirla, agregue su ruta a la variable de entorno:
2. Instalar dependencias:
instalación de conda -c fvcore -c iopath -c conda-forge fvcore iopath
3. Seleccione la versión de Pytorch3d. Cada versión de Pytorch3d tiene su versión de pytorch aplicable indicada, como se muestra a continuación:
La imagen de arriba muestra que las versiones de Pytorch aplicables de pytorch3d 0.6.2 son 1.10.1, 1.10.2, 1.11.0 y Python aplicable es 3.10, pero Pytorch1.6.0 y Python3.6 ya no son compatibles. torch aún sigue esto Descargue la versión pytorch3d correspondiente
4. Después de instalar VS 2019, abra el terminal "Símbolo del sistema de herramientas nativas x64 para VS 2019" como se muestra a continuación, y luego acceda a la ruta del directorio donde se descomprime pytorch3d:
Ingrese en orden:
Establezca variables de entorno en el símbolo del sistema en Windows:
establecer DISTUTILS_USE_SDK=1
establecer PYTORCH3D_NO_NINJA=1
Establezca variables de entorno en PowerShell en Windows:
$env:DISTUTILS_USE_SDK = 1Establezca variables de entorno en Linux y macOS:
export DISTUTILS_USE_SDK=1
Luego, modifique el código fuente del archivo setup.py en pytorch3d: cambie extra_compile_args = {“cxx”: [“-std=c++14”]} a: extra_compile_args = {“cxx”: []};
5. Finalmente ingrese en la terminal "Símbolo del sistema de herramientas nativas x64 para VS 2019":
python setup.py instalar --usuario
El proceso de instalación tomó aproximadamente 5 minutos en mi computadora y la página siguió saltando de un lado a otro hasta que finalmente mostró éxito:
Si su instalación aún falla, por favor contácteme