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版本。