前言:
nvidia在github上发布的flownet2-pytorch并不能按照它说明的那样直接安装,里面存在很多bug, 根据上网查询与自己实践,我把安装步骤一步一步列出来,供大家参考。
1) 创建环境
conda create -n flownet2 python=3.6.9
2)进入虚拟环境
conda activae flownet2
3)安装gcc7,g++7
sudo add-apt-repository ppa:ubuntu-toolchain-r/test
sudo apt-get update
sudo apt-get install gcc-7
sudo apt-get install g++-7
4)将gcc7,g++7作为默认选项
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 100
sudo update-alternatives --config gcc
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-7 100
sudo update-alternatives --config g++
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 50
5) 查看所有gcc版本
ls /usr/bin/gcc*
6) 查看当前gcc版本
gcc -v
7)安装pytorch
下面介绍的可行版本1和可行版本2是我自己测试过的,可行版本3是评论区反馈后测试过的。
可行版本1(ubuntu 16.04, cuda 10.2, NVIDIA Driver: 440.36,gcc 5.4.0):
conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=10.1 -c pytorch
可行版本2(ubuntu 16.04, cuda 10.2,NVIDIA Driver: 440.36,gcc 5.4.0):
conda uninstall python
conda install python==3.7
conda install pytorch==1.9.0 torchvision==0.6.1 cudatoolkit=10.2 -c pytorch
可行版本3(ubuntu 18.04, cuda 10.1,GCC 7.4):
conda uninstall pytho

这篇博客详细记录了在Ubuntu系统中安装NVIDIA的Flownet2-PyTorch库的过程,包括创建环境、安装依赖、解决编译错误和测试流程。博主提到了安装过程中遇到的gcc版本问题,以及pytorch版本与CUDA版本的适配。还分享了在代码运行时可能遇到的模块导入错误及其解决方案,并提供了测试数据集和checkpoint的下载链接。
最低0.47元/天 解锁文章
1918





