安装torch(非pytorch)
系统环境ubuntu16.04+cuda9.0+gcc5.0+cudnn7
*cuda9只对应cudnn6以上版本,但是torch默认支持cudnn5.x版本,所以会出现版本不兼容问题,后面会写如何解决,如果不想出现该问题,可以直接安装cuda8.0+cudnn5(没试过,应该可以)
git clone https://github.com/torch/distro.git ~/torch –recursive
cd ~/torch
bash install-deps
export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__"
bash install.sh
安装编译需要一会时间。
安装好后,会提示设置环境变量,输入yes就行。
然后执行source ~/.bashrc
激活相应的环境变量。
这是在终端输入th就可以呼出torch的控制界面:
注意事项
1、torch默认支持cudnn5,使用cudnn7会报如下错误