Windows 安装 pytorch3d

Windows 安装 pytorch3d

首先安装pytorch，我安装的是v0.2.5

python 3.8 + cuda 10.2 + pytorch 1.7.1 + torchvision 0.8.2 + cub 1.10.0

其中，cudnn和cudatoolkit与cuda的版本一致的，cub直接conda是不行的，需要从github上下载release，再在环境变量里添加CUB_HOME

作者：胡写含
链接：https://www.jianshu.com/p/ffd696866469
来源：简书
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

pytorch                   1.5.0           py3.7_cuda102_cudnn7_0    pytorch

然后下载pytorch3d到本地，主要不要下载master分支的，下载release的版本 ，我这里下载最新的 v0.2.0，下载链接是 https://github.com/facebookresearch/pytorch3d/archive/v0.2.0.zip.

下载到本地解压后，再 以管理员方式打开 x64 Native Tools Command Prompt for VS 2019

再进入pytorch3d目录

cd pytorch3d-0.2.0
activate conda_enviroment
python setup.py install

等待片刻，便能成功安装！