CUDA10安装支持gpu的tensorflow版本

本文详细介绍了如何在CUDA10环境下安装并配置支持GPU的TensorFlow1.12版本,包括下载相关依赖包及具体安装步骤,确保深度学习模型训练的顺利进行。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

由于英伟达官方宣称CUDA10有着巨大的性能提升,我便迫不及待的想尝试用它来训练深度学习模型,然而现在Tensorflow的最新官方GPU版本1.12并不支持在CUDA10下的模型训练,因此便翻墙到国外论坛找材料,最终找到了可以完美适配CUDA10的tensorflow方案。具体的环境如下:


安装步骤:

首先将相关的轮子依赖下载到本地系统中,(https://download.csdn.net/download/weixin_38538282/10816219或者翻墙下载:https://1drv.ms/u/s!AspzPyzQ0HQR7VNCicPlnCV2abKJ)我选择的是Pycharm工程venv目录下的Scripts,然后再进入到该目录下,输入pip3 install tensorflow-1.12.0rc0-cp36-cp36m-win_amd64.whl 命令,即可成功安装CUDA10下支持的tensorflowGPU版本。

验证:进入python

输入:pip install tensorflow 不报错即可


 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值