本文主要是介绍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
这篇关于Pytorch添加自定义算子之(1)-安装配置Eigen库的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!