FlowNet2安装踩雷经验
服务器重启之后硬盘被清空,需要重新配置环境,之前就在安装FlowNet2时碰到不少麻烦,这回又卡了一小会。发现中文网站上对Flownet安装中的指南不多,遂决定记录一下。
首先我的环境是Ubuntu18.04,cuda10.0 pytorch1.4
第一个问题是在python3.7的环境下安装Flownet
get flownet2-pytorch source git clone
https://github.com/NVIDIA/flownet2-pytorch.git
cd flownet2-pytorchinstall custom layers
bash install.sh
报错
correlation_cuda.cc:4:10: fatal error: ATen/cuda/CUDAContext.h: No such file ordirectory
出错在cuda的cpp文件上,由于对cpp不熟,没法修改cpp文件。谷歌上搜了一下,github上不少人遇到这个问题,时间紧迫懒得细究原因,根据底下老哥摸索出来的配置
pytorch 0.4.1, cuda 8/9, python36 nopes
pytorch 1.0.0, cuda 10, python36 works
pytorch 1.0.1, cuda 10, python36 nopes
pytorch 1.1.0, cuda 10, python36 works
重新基于python3.6配置了环境,成功安装Flownet2