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
等待片刻,便能成功安装!