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)