在windows10上安装caffe和tensorflow

最近在Windows10上安装了caffe和tensorflow,折腾了好久。在此记录一下。

安装caffe的过程已在另一篇博客中进行了记录,在此不再赘述。而tensorflow也是非常简单的,也不再详细说明。

安装caffe和tensorflow比较重要的一点是,要确保把依赖正确安装好。

GPU版本的caffe需要7+版本的CUDA和5.5/5.0的CUDNN,比较常见的组合是CUDA8.0搭配CUDNN5.0。而GPU版本的tensorflow在1.5.0以上的版本则需要CUDA9和CUDNN7。两个框架对CUDA和CUDNN的不同要求造成了小小的冲突。

因为caffe需要的CUDA和CUDNN的版本较低,所以需要先安装好低版本的CUDA和CUDNN,把caffe搭建好。然后再安装高版本的CUDA和CUDNN,安装tensorflow。不同版本的CUDA是可以共存的,因为不同程序可以链接不同版本的CUDA文件。

caffe和tensorflow对python的版本也有要求,使用Anaconda管理python会比较方便,通过创建虚拟环境可以更改python的版本。

猜你喜欢

转载自www.cnblogs.com/Peyton-Li/p/10688106.html