TensorFlow 1.2 依赖cuda8 + cudnn5.1
Pytorch则要求cudnn ≥ 6
需要做的操作:
(1)先安装cudnn5.1,再安装cudnn6,cuda/lib64文件夹下的libcudnn库关系如下:
(2)重装TensorFlow
TensorFlow 1.2 依赖cuda8 + cudnn5.1
Pytorch则要求cudnn ≥ 6
需要做的操作:
(1)先安装cudnn5.1,再安装cudnn6,cuda/lib64文件夹下的libcudnn库关系如下:
(2)重装TensorFlow