Ubuntu下源码安装Pytorch

Ubuntu下源码安装Pytorch

---

环境准备

1. 安装CUDA , 下载地址NVIDIA 官网， 安装教程推荐博客
2. 安装 Anaconda python 环境

    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

1. 下载源码， 并安装

git clone --recursive https://github.com/pytorch/pytorch
cd pytorch