Reference:
1. 光流估计——从传统方法到深度学习
2. Flownet2 NVIDIA pytorch最新安装教程 有效的避坑教程
3. 使用NVIDIA flownet2-pytorch实现生成光流
一、环境配置
默认背景:已安装好Anaconda3,cuda 10.0,cuDnn 7.6.0
- 安装Anaconda3:镜像下载地址
- 修改jupyter notebook默认工作路径:
-
- anaconda prompt 中
jupyter notebook --generate-config
生成配置文件
- anaconda prompt 中
-
- 打开这个配置文件,修改notebook_dir为指定路径,
c.NotebookApp.notebook_dir = 'D:\ForPython\workplace'
- 打开这个配置文件,修改notebook_dir为指定路径,
-
- jupyter notebook右键属性-快捷方式-目标,将最后的
%USERPROFILE%
删掉即可
- jupyter notebook右键属性-快捷方式-目标,将最后的
1. 下载flownet2-pytorch项目
2. 安装pytorch
- 打开Anaconda Prompt,创建自定义名为pytorch的环境,Python3.5版本
conda create -n pytorch python=3.5
- 激活自定义pytorch环境
conda activate pytorch
版本根据情况选择:win10 + cuda10 + python3.5 对应下载
- torch-1.0.1-cp35-cp35m-win_amd64.whl
- torchvision-0.4.0-cp35-cp35m-win_amd64.whl