仅作为个人学习记录使用
环境:Ubuntu20.04.4 LTS
我遇到的问题:为ATen算子添加函数符号,首先需要获取ATen中算子接口定义,torch/_C/_VariableFunctions.pyi
和 torch/nn/functional.pyi
这两个文件中搜索我们刚刚得到的这个算子名。
这两个文件是编译 PyTorch 时本地自动生成的文件,里面包含了 ATen 算子的 PyTorch 调用接口。
那么我需要先找到当前容器使用的torch包安装位置
安装位置如下:
admin:///var/lib/docker/overlay2/f1e61496c4be45ab3eda590df171a9237c12e75ec05f282c43c0f2230d697a3e/merged/opt/conda/envs/deploy/lib/python3.7/site-packages/torch/
其他python包也在site-packages文件夹下
docker的工作目录:
admin:///var/lib/docker/overlay2/f1e61496c4be45ab3eda590df171a9237c12e75ec05f282c43c0f2230d697a3e/merged/root/workspace
Ubuntu下Docker安装在var/lib/docker
/文件夹下