电脑配置:ubuntu16.04 cuda8.0 cudnn6 anaconda虚拟环境 python2.7
- 配置依赖 ,设置cuda
export CMAKE_PREFIX_PATH="$(dirname $(which conda))/…/" # [anaconda root directory]
conda install numpy pyyaml mkl mkl-include setuptools cmake cffi typing
conda install -c mingfeima mkldnn
conda install -c pytorch magma-cuda80 - 下载pytorch 源码
git clone --recursive https://github.com/pytorch/pytorch.git
cd pytorch - 运行安装
python setup.py install