tensorflow-gpu安装
由于最近在学习tensorflow,需要在windows环境下安装tensorflow,刚开始装出现了各种报错,基本都是版本不对应,后来重新安装一遍,可以使用了,这里记录一下。
我没有使用anaconda安装python,直接使用pip进行安装的。
- 首先win+R,运行cmd,运行如下命令
pip install tensorflow-gpu==2.3.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
我这里使用的是python 3.7版本,安装的是tensorflow-gpu的2.3.0版本。
- 接下来需要安装CUDA和CuDNN,这里需要选择对应的安装版本,版本对应的关系如下图(图片来自于水印)
- 先安装CUDA,到其官网https://developer.nvidia.com/cuda-downloads,选择对应的10.1版本安装,
然后下载完成后,一路Next安装就可以了。
- 安装完成后,需要将CUDA对应的路径配置到环境变量,我的默认安装路径为:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1,分别划分红框中的三个以及根路径添加进环境变量;
- 接下来需要下载CuDnn,到官网https://developer.nvidia.com/cudnn,选择对应的7.6版本下载,下载时需要进行一个简单的注册和调研,下载后进行解压,
- 解压后,找到cuda安装目录中对应的三个文件夹,分别将cudnn中的三个文件夹内容复制(注意这里不是直接复制文件夹,要进入文件夹,复制dll文件),然后粘贴到对应的cuda目录中的三个文件夹中。
安装完成后就基本完成了,打开cmd,运行nvcc -V查看cuda安装版本:
nvcc -V
运行nvidia-smi,查看显卡信息和gpu进程:
在python 命令行输入:
import tensorflow as tf
print(tf.test.is_gpu_available())
则显示 gpu可用,说明安装成功。
分类:
日常随笔
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧