flownet2-pytorch issues

GitHub：https://github.com/NVIDIA/flownet2-pytorch
———————————————————————————————————————————————

pytorch>=0.4.1

1.bash ./install.sh报错：nvcc fatal : Unsupported gpu architecture ‘compute_70’

• cuda版本要求9.0
  1> pytorch安装对应版本
  conda install -n py36t4 pytorch=0.4.1 torchvision cudatoolkit=9.0 -c pytorch
  2>查看nvcc版本
  nvcc --version
  3>修改当前cuda（服务器）
  • vim ~/.bash.profile

# .bash_profile
  2
  3 # Get the aliases and functions
  4 if [ -f ~/.bashrc ]; then
  5     . ~/.bashrc
  6 fi
  7
  8 # User specific environment and startup programs
  9
 10 PATH=/usr/local/cuda2/bin:$HOME/.local/bin:$HOME/bin:$PATH
 11 export PATH
 12
 13 LD_LIBRARY_PATH=/usr/local/cuda2/lib64:$LD_LIBRARY_PATH
 14