TensorFlow2.16.1-gpu (windows10 wsl2 ubuntu)
1，官网引导2，原生widows上gpu支持到2.10版本
原生windows gpu支持
Tensorflow-gpu保姆级安装教程https://csdnimg.cn/release/blog_editor_html/release2.3.6/ckeditor/plugins/CsdnLink/icons/icon-default.png?t=N7T8https://blog.csdn.net/weixin_43412762/article/details/129824339
3，windows想要安装最新版的TensorFlow2.16.1 gpu支持 必要wsl2。
windows本机安装
Official Drivers | NVIDIA 安装本身GPU适配的最新版
CUDA Toolkit Archive | NVIDIA Developer 最新版
CUDA Deep Neural Network (cuDNN) | NVIDIA Developer 最新版
https://img-blog.csdnimg.cn/direct/23d9100d77644390a14bbba2fdfa38d5.jpeg
windows wsl2
https://img-blog.csdnimg.cn/direct/b7b932774d46414ebc5b1014ea39421a.jpeg
windows10上wsl 旧版 WSL 的手动安装步骤 | Microsoft Learn
安装好后在PowerShell和在wsl中使用如下命令输出信息一样
nvidia-smi 4，进入wsl中安装tf2.16.1
pip安装换国内源，下面是阿里云的示例
pip install -i https://mirrors.aliyun.com/pypi/simple/ --upgrade pip
pip install -i https://mirrors.aliyun.com/pypi/simple/ tensorflow
5，GPU not detected on WSL2 · Issue #63341 · tensorflow/tensorflow · GitHub
https://img-blog.csdnimg.cn/direct/c9e5d0f80cf54ac7bf0f733d2cb2830c.jpeg
按照上图的步骤操作就能乐成安装tf2.16.1-gpu
最新版的12.4现在还不支持
一定要 Cuda toolkit 12.3 和配套的 Cudnn 8.9 选择下载wsl对应版本的安装包
https://img-blog.csdnimg.cn/direct/d1552e3ee6094bc2ac9a87cd6d8914db.jpeg
cudnn8.9安装参考最新版9.1.1的安装步骤，把包名替换成8.9的
https://img-blog.csdnimg.cn/direct/97e2d6c824cf476b96767229d82083d4.jpeg
假如安装Cuda toolkit12.3前已经安装了12.4，要完全卸载再安装12.3。cuda卸载
假如安装cudnn8.9之前已经安装了cudnn9，起首 完全卸载cudnn9
安装cudnn8.9的上图步骤实验到 apt install cudnn 的时间会发现安装的还是cudnn9，使用 apt search cudnn 检察
完全卸载cudnn9后，使用如下命令安装cudnn8
sudo apt install libcudnn8 libcudnn8-dev -y
#apt search cudnn8 #查看是否安装成
#find /usr -name '*cudnn*' #查看cudnn头文件和库路径 设置情况变量
vim ~/.bashrc
#在脚本末了添加以下两行
export PATH=/usr/local/cuda-12.3/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-12.3/lib64:$LD_LIBRARY_PATH
#保存退出
source ~/.bashrc
完成以上全部步骤，实验如下python语句
python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
https://img-blog.csdnimg.cn/direct/3d5d61f3d2b044069ba110d98542b01b.jpeg
忽略警告。tf2.16.1-gpu安装乐成。
GPU not detected on WSL2 #63341 还提到了的其他方法，可自行尝试。
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。
页:
[1]