Pytorch添加自定义算子之（1）-安装配置Eigen库

一、安装对应的ubuntu环境

推荐使用Docker

FROM nvcr.io/nvidia/pytorch:23.01-py3
RUN pip install tensorboardX
RUN pip install pyyaml
RUN pip install yacs
RUN pip install termcolor
RUN pip install opencv-python
RUN pip install timm==0.6.12
WORKDIR /app
COPY . /app

二、下载对应的Eigen

git clone https://gitlab.com/libeigen/eigen.git
cd eigen/
mkdir build && cd build
cmake ..

make install