einops
前言
如果要跑Vision Transformer,仅仅有pytorch是不够的,还需要一个工具包einops,笔者在配置该环境时,没有看到直接的教程,谨以本文进行记录。
步骤
1. 在terminal
中输入conda info --envs
,可以查看当前你的anaconda中建立的各个环境。
2. 使用命令 source activate openCV
进入名为openCV
的环境中。
(这里使用 conda list
命令可以查看当前环境下安装的包)
3. 使用pip命令进行第三方库的下载
注意:如果使用conda 需要去配置channels,不然会显示找不到。
4. 再次使用conda list
就可以看到有einops啦
参考博客:Anaconda环境下的基本操作
学习博客:einops README: 如何优雅地操作张量维度
einops张量操作神器