当前位置:   article > 正文

Tensorflow1.14 GPU版本环境配置复盘_tensorflow1.14环境配置

tensorflow1.14环境配置

Tensorflow1.14 GPU版本环境配置复盘

注意:Tf1.15版本之前GPU和CPU的软件包是分开的

Tensorflow1.x GPU版本的安装并不像pytorch,使用conda下载tensorflow时,包里面并不包含对应的cuda。而且在我们的电脑中,其实本来并没有安装cuda,所以要提前配置。

一些定义:

Cudnn:NVIDIA cuDNN(CUDA Deep Neural Network)是用于深度神经网络的GPU加速库。

Cuda:NVIDIA公司发布了CUDA(Compute Unified Device Architecture),是一种新的操作GPU计算的硬件和软件架构,是建立在NVIDIA的GPUs上的一个通用并行计算平台和编程模型,它提供了GPU编程的简易接口

从上面可以看出我们要下载cuDNN 7.4;CUDA 10;python 3.5-3.7。

(11条消息) win10系统CUDA10.0安装教程(for tensorflow2.0)_师范大学生的博客-CSDN博客_cuda10.0

CUDA安装参照上文。注意,一个电脑是可以安装多个CUDA的,这里的问题是不同版本CUDA的文件名字是一模一样的。并且我们在调用CUDA时需要配置环境变量(环境变量配置完需要重启),所以可以通过环境变量的顺序改变,让程序找到不同的cuda版本。

 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/凡人多烦事01/article/detail/171319
推荐阅读
相关标签
  

闽ICP备14008679号