在Google Colab中配置RAFT光流模型conda环境遇阻求助

我是一名学生，第一次用Stack Overflow。目前在做学校项目，基于普林斯顿视觉实验室的RAFT光流模型，想在Google Colab上运行，但卡在了conda环境配置和预训练模型加载这两步。这是我第一次做这类操作，完全摸不着头绪，希望有人能帮忙。

我尝试的代码如下：

# conda install
!wget -c https://repo.continuum.io/archive/Anaconda3-5.1.0-Linux-x86_64.sh
!chmod +x Anaconda3-5.1.0-Linux-x86_64.sh
!bash ./Anaconda3-5.1.0-Linux-x86_64.sh -b -f -p /usr/local

# conda update to avoid inconsistencies
!conda install anaconda
!conda update --all
!conda update conda -y -q;;
!source /usr/local/etc/profile.d/conda.sh
!conda init bash
!conda install -n root _license -y -q

# raft setup
!conda create --name raft
!conda activate raft
!conda install pytorch=1.6.0 torchvision=0.7.0 cudatoolkit=10.1 matplotlib tensorboard scipy opencv -c pytorch

我参考的是普林斯顿视觉实验室RAFT仓库的说明文档。之前试过找其他能用conda在Colab安装的光流模型代码，把模型换成RAFT（而非FlowNet或GMA），但没成功；也单独装过CUDA，结果遇到conda环境不一致的问题，所以加了conda更新的代码，还是不行。

内容的提问来源于stack exchange，提问作者SSH