1 2 3 4

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,这里需要选择对应的安装版本,版本对应的关系如下图(图片来自于水印)

然后下载完成后,一路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可用,说明安装成功。

posted @   Uniqe  阅读(180)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
点击右上角即可分享
微信分享提示