手把手安装flownet2-pytorch

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

前言:

        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
评论 20
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值