pytorch3d windows安装坑注意点

pytorch3d在windows上的安装真的是个大坑,踩完坑来分享一下经验。

首先是winodows上的pytorch3d不能通过conda直接安装,而是需要下载源代码执行相关脚本安装,具体见:https://www.jianshu.com/p/0e798b29d39e

关于RuntimeError: Error compiling objects for extension这个问题,主要应该是出在Pytorch版本以及cuda版本的问题上。
经过我的尝试,最终使用 pytorch1.6.0 cuda 10.2 是没有问题的

conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.2 -c pytorch

你可能感兴趣的:(pytorch踩坑,人工智能,python)