0
点赞
收藏
分享

微信扫一扫

tensorflow-gpu安装及注意事项

infgrad 2022-01-21 阅读 97
tensorflow

tensorflow-gpu安装

1.CUDA安装

打开显卡驱动,找到显卡对应的CUDA版本(如图所示 此电脑显卡对应的cuda为11.3.1)
帮助-》系统信息-》组件
在这里插入图片描述
如图所示,本电脑的CUDA版本为11.3.1

1.1下载CUDA(https://developer.nvidia.com/cuda-toolkit-archive)

进入CUDA官网,选择对应的CUDA版本
在这里插入图片描述
解压,安装,安装时可自定义路径。

1.2CUDA配置环境变量

在这里插入图片描述
输入nvcc -V 出现如图信息,安装成功
在这里插入图片描述

2.安装cudnn(https://developer.nvidia.com/rdp/cudnn-archive)

下载cudnn之前需要注册账号,按照要求填信息即可,注册成功后即可下载。
选择CUDA版本对应的cudnn版本
在这里插入图片描述
将下载好的文件,解压。将里面三个文件夹(bin、include、lib)下的文件,复制到CUDA安装路径下的的同名文件中。
在这里插入图片描述
在这里插入图片描述

3.tensorflow-gpu安装

3.1查看CUDA版本对应的tensorflow版本(这里没有显示11.3.1对应的tensorflow版本,我这里选择安装2.6的版本也是可以的)(https://www.tensorflow.org/install/source_windows)

在这里插入图片描述
使用anaconda安装对应的tensorflow-gpu版本
在这里插入图片描述

3.2测试是否安装成功

import tensorflow as tf
tf.config.list_physical_devices('GPU')

在这里插入图片描述

4.参考链接

https://blog.csdn.net/sinat_23619409/article/details/84202651

https://blog.csdn.net/sunmingyang1987/article/details/102872658

举报

相关推荐

0 条评论