win11+rtx3060ti+tensorflow安装

win11+rtx3060ti+tensorflow安装

一、查看rtx3060ti对应的cuda版本

1、打开英伟达控制面板

开始菜单，搜索英伟达控制面板，帮助—系统信息，组件，查看cuda版本

由以上图片可以看出，本机当前cuda11.6

二、安装cuda

1、进入cuda官网

cuda 官网： https://developer.nvidia.com/cuda-toolkit-archive
选择对应的cuda版本，并下载！

如本机，选择windows，x86_64 (64位)，version11,exe(local),先下载到本地，再安装

2、安装cuda

解压，双击安装

三、安装cudnn

1、下载cudnn

进入cudnn官网，https://developer.nvidia.com/cudnn
选择与cuda版本匹配的cudnn,并下载，这里需要注册认证，免费的，但是需要填一些资料

2、安装cudnn

将下载好的cudnn解压，复制里面的 bin include lib 三个文件夹的内容，粘贴到cuda安装目录下

若忘记了cuda的安装路径，可以cmd，set cuda

如图，本机应该将cudnn的三个文件夹，复制到 v11.6的目录下

四、安装tensorflow

本机使用的pycharm编译器，直接在pycharm里安装的，使用命令行，此处需要注意，最新的tensorflow2.11的GPU不可用，故需要使用2.10版本
pip install tensorflow==2.10 -i https://pypi.douban.com/simple/

五、 测试

测试GPU是否可用！

	import tensorflow as tf
	version = tf.__version__
	gpu_ok = tf.config.list_physical_devices("GPU")
	print("tf version:",version,"\nuse GPU:",gpu_ok)
	cuda = tf.test.is_built_with_cuda()
	print("cuda:",cuda)