参考官网: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
没有安装成功,后面补充。