Linux下源码编译libtorch

参考官网:https://github.com/pytorch/pytorch/tree/v1.5.1

尝试验证。

安装依赖:

conda install numpy ninja pyyaml mkl mkl-include setuptools cmake cffi
conda install -c pytorch magma-cu102

安装:

export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which conda))/../"}
python setup.py install

没有安装成功,后面补充。

猜你喜欢

转载自blog.csdn.net/juluwangriyue/article/details/108607842
今日推荐