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

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

 

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值