FlowNet2安装踩雷经验

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-pytorch

install 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

已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 1024 设计师:白松林 返回首页