Reference official website: https://github.com/pytorch/pytorch/tree/v1.5.1
Try to verify.
Installation dependencies:
conda install numpy ninja pyyaml mkl mkl-include setuptools cmake cffi
conda install -c pytorch magma-cu102
installation:
export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which conda))/../"}
python setup.py install
The installation was not successful, please add later.