mvcnn-pytorch版代码:https://github.com/jongchyisu/mvcnn_pytorch
环境中要安装的内容:
python 3.6
pytorch 0.4.1 (记得装gpu版)
torchvision 0.2.2
cudatoolkit 9.0
cudnn 7.6.5
scikit-image 0.13.0(conda自动给我选的)
tensorboardX 1.4
由于这个方法比较老了,用的库版本也比较老,现在的设备配起来不那么方便。这里注意:
pytorch 0.4.1需要下载.whl文件手动安装,在这个网站里https://download.pytorch.org/whl/torch_stable.html找到cu开头的,自己需要的版本的文件,比如我需要配cuda9.0,在win10系统上,就下载torch-0.4.1-cp36-cp36m-win_amd64.whl,这个cp后面的号码我不懂所以就按python版本设想了。下好用pip install 本地安装就好了。
python, cudatoolkit, cudnn, scikit-image,tensorboardX可以用conda install正常装,其中tensorboardX用官网给的方式装https://anaconda.org/conda-forge/tensorboardx。torchvision也用pip装,用我前一条博文里的指令。
我使用的系统是win10系统,在运行时会报错:ValueError: ‘modelnet40_images_new_12x’ is not in list,解决方案可参考:
https://github.com/jongchyisu/mvcnn_pytorch/issues/29 中1987553369的回答