Win10下配置tensorflow-gpu(无VS2015/2017)

本文介绍了在Windows 10操作系统中,如何在没有Visual Studio 2015/2017的情况下配置tensorflow-gpu。步骤包括确定tensorflow-gpu版本、更新NVIDIA驱动、安装CUDA,设置环境变量,下载并放置cudnn文件,使用conda创建虚拟环境,安装tensorflow-gpu,最后验证安装是否成功。

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

总体顺序

  1. 确定需要安装的tensorflow-gpu版本,点击这里拉到最下方,一般是cuda10和cudnn7.4,以及对应的nvidia驱动,cuda,cudnn版本
  2. 更新驱动,可下载NVIDIA-GEFORCE,点击这里
  3. 安装cuda,点击这里最后选择exe(local)
  4. 下载完成后选择一下安装目录,其他的不用管(记住自己的安装目录)
  5. 添加cuda环境变量,自行百度
  6. 下载cudnn7.4,nvidia官网注册账号后下载,百度一下将cudnn哪三个包放到对应cuda文件夹下即可
  7. 安装anaconda,打开cmd,创建虚拟环境,方便管理不同版本,比如conda create -n tf20 python=3.6
  8. 最后一步,打开cmd,换个清华镜像或者中科大镜像,自行百度,然后pip install tensorflow-gpu==2.xx.x(版本号不对就从提示中选择一个正确的)
  9. 打开一个python文件,在setting中选择anaconda3,envs,tf20,python.exe 在这里插入图片描述

最后一步

import tensorflow as tf
print(tf.__version__)

输出是这样子的,就是成功了

2020-07-12 11:23:01.773201: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_100.dll
2.0.0

如果想查看运行状态,任务管理器,性能,在GPU下边有个3D,点击3D的小箭头,选择cuda,gpu跑起来cuda会占满。
GPU可能跑的时候占用不高,cuda占用高就行

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值