怎样安装tensorflow?

1.安装cuda

表明我的电脑中没有cuda

2.我应该安装什么版本的cuda呢?

 

查找到驱动版本的程序信息,我的显卡驱动版本是512.13.

去英伟达官网查找能安装的cuda版本

https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html 

查找到自己能安装的cuda版本为11.6多。

3.怎样安装cuda?

cuda官网下载,https://developer.nvidia.com/cuda-toolkit-archive

自定义安装

没有使用visual Studio,所以不能勾选

 然后安装要等待一段时间

显示环境变量已经配置好了

检查cuda是否安装好

查看cuda设置的环境变量

找到cuda的安装目录,找到“nvcc.exe”

4.cuDNN神经网络加速库安装

cuDNN Archive | NVIDIA Developerhttps://developer.nvidia.com/rdp/cudnn-archive

 

把这四个文件复制到该文件夹下,只需要强制粘贴

 

cuDNN只是cuda的一个补丁,优化深度学习用的。

配置环境变量,只需要把粘贴好的四个文件所在路径创建环境变量。

 5.检查cuDNN是否安装好?

怎样在命令行打开指定文件夹?

因为这个C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\extras\demo_suite路径好长,直接找到文件所在位置,在路径栏里输入cmd,就可以在cmd中打开这个文件夹。

 

两个pass表明cuDNN配置成功。

 

6.tensorflow的GPU安装

命令行输入pip install -U tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple

出现这个提示:[notice] To update, run: python.exe -m pip install --upgrade pip
 

下面警告好像是说我C盘里有python.exe,我也去运行下。

 已经安装过了,没报错我就不改了。

 检查tensorflow-gpu是否安装成功?

在python环境下输入如下语句,如果下面显示True就表示安装成功。

 查看服务器类型

tensorflow的版本信息:

 

到这里TensorFlow就安装成功了。 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值