在Win11下安装pytorch3d

原创已于 2024-02-26 20:12:45 修改 · 1.8k 阅读

16 ·

CC 4.0 BY-SA版权

文章标签：

#AI

于 2024-02-22 11:29:38 首次发布

AI 专栏收录该内容

89 篇文章

订阅专栏

本文详细描述了如何在Windows系统上安装MingW64编译器，VisualStudio2019开发环境，以及如何通过Conda管理工具安装PyTorch及其相关依赖项，包括PyTorch3D的源代码编译过程和CUDA版本检查。

部署运行你感兴趣的模型镜像

安装MingW64

MinGW-w64 - for 32 and 64 bit Windows download | SourceForge.nethttps://sourceforge.net/projects/mingw-w64/选择x86_64-win32-seh，下载解压后将目录添加到环境变量Path中。

安装Visual Studio 2019

安装的时候一定要选择language C++。

安装成功后，将cl.exe的路径添加到环境变量Path中。

一般cl.exe的地址是在 C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Tools\MSVC\***\bin\Hostx64\x64 里面。

安装依赖项

conda create -n pytorch3d python=3.10
conda activate pytorch3d
conda install pytorch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 pytorch-cuda=11.8 -c pytorch -c nvidia
conda install -c fvcore -c iopath -c conda-forge fvcore iopath

安装pytorch3d

下载pytorch3d源码

git clone https://github.com/facebookresearch/pytorch3d.git

以管理员身份打开“x64 Native Tools Command Prompt...”

在conda环境下，进入pytorch3d目录执行编译

conda activate pytorch3d
cd pytorch3d
set DISTUTILS_USE_SDK=1
python setup.py install

检查cuda版本

import torch
torch.cuda.is_available()

检测是否安装成功：

import pytorch3d
print(pytorch3d.__version__)

您可能感兴趣的与本文相关的镜像

PyTorch 2.9

PyTorch

Cuda

PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理