git clone –recursive https://github.com/pytorch/pytorch

``` python setup.py install ``` 如果需要限制CMAKE的cpu数量为2，使用```MAX_JOBS=2 python setup.py install```命令进行编译安装。

若出现：```ptxas warning : Too big maxrregcount value specified 96, will be ignored```问题， 将/third_party/nccl/nccl/makefiles/common.mk第28行位置处sm_30的选项去掉： ``` … CUDA8_GENCODE = -gencode=arch=compute_30,code=sm_30 \

1. gencode=arch=compute_35,code=sm_35 \
2. gencode=arch=compute_50,code=sm_50 \
3. gencode=arch=compute_60,code=sm_60 \
4. gencode=arch=compute_61,code=sm_61

… 修改为： … CUDA8_GENCODE = -gencode=arch=compute_35,code=sm_35 \

1. gencode=arch=compute_50,code=sm_50 \
2. gencode=arch=compute_60,code=sm_60 \
3. gencode=arch=compute_61,code=sm_61

… ```