Ubuntu下源码安装PyTorch
环境准备
export CMAKE_PREFIX_PATH="$(dirname $(which conda))/../" # [anaconda root directory]
# Install basic dependencies
conda install numpy pyyaml mkl mkl-include setuptools cmake cffi typing
# Add LAPACK support for the GPU
conda install -c pytorch magma-cuda80 # or magma-cuda90 if CUDA 9
- 下载源码, 并安装
git clone --recursive https://github.com/pytorch/pytorch
cd pytorch