Tensorflow 和 pytorch 依赖cudnn版本的冲突问题

前端之家收集整理的这篇文章主要介绍了Tensorflow 和 pytorch 依赖cudnn版本的冲突问题前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

TensorFlow 1.2 依赖cuda8 + cudnn5.1

Pytorch则要求cudnn ≥ 6

需要做的操作:

(1)先安装cudnn5.1,再安装cudnn6,cuda/lib64文件夹下的libcudnn库关系如下:


(2)重装TensorFlow

原文链接:https://www.f2er.com/javaschema/282335.html

猜你在找的设计模式相关文章