Tensorflow-gpu下载

原创已于 2024-03-16 17:13:23 修改 · 2.1k 阅读

47 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow #学习 #python

于 2024-03-16 17:10:32 首次发布

本文讲述了作者在安装Tensorflow-gpu时遇到的问题，包括CUDA版本不匹配、CUDNN缺失以及DLL文件的加载错误。文章详细描述了如何检查CUDA和CUDNN版本，下载相关库，以及解决DLL文件缺失的方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Tensorflow-gpu下载

下载
验证
问题

下载

本机是CUDA 12.3 但是下载安装的是v12.0
cudnn 8.9.4.25
tensorflow-gpu 2.6.0
python 3.9

cuda下载地址：CUDA Toolkit Archive | NVIDIA Developer；
cudnn下载地址：cuDNN Archive | NVIDIA Developer。

我的安装过程是这样的

1 查看CUDA版本

在这里插入图片描述

2 CUDA 和 cdunn,版本对应…，

这篇博客写的真好
https://blog.csdn.net/weixin_43412762/article/details/129824339

下载cudnn: https://developer.nvidia.com/rdp/cudnn-archive
在这里插入图片描述

3 安装tensorflow-gpu

下载tensorflow的whl文件直接安装：https://pypi.tuna.tsinghua.edu.cn/simple/tensorflow-gpu/

pip install C:\Users\shaoj\Downloads\tensorflow-2.6.0-cp39-cp39-win_amd64.whl

直接下载：

pip install tensorflow=2.6.0

验证

import tensorflow as tf
print(tf.__version__)
print(tf.test.is_gpu_available())

问题

报错缺少dll文件
2024-03-16 16:57:07.173514: W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library ‘cublas64_11.dll’; dlerror: cublas64_11.dll not found
2024-3-16 16:57:07.197352: W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library ‘cublasLt64_11.dll’; dlerror: cublasLt64_11.dll not found
2024-3-16 16:57:07.206373: W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library ‘cufft64_10.dll’; dlerror: cufft64_10.dll not found
2024-03-16 16:57:07.229122: W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library ‘curand64_10.dll’; dlerror: curand64_10.dll not found
2024-03-16 16:57:07.246726: W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library ‘cusolver64_10.dll’; dlerror: cusolver64_10.dll not found
2024-03-1620:16:07.254984: W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library ‘cusparse64_11.dll’; dlerror: cusparse64_11.dll not found
2024-03-16 16:57:07.261778: W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library ‘cudnn64_8.dll’; dlerror: cudnn64_8.dll not found

电脑里全局搜索缺少的dll文件
在这里插入图片描述

将其复制进CUDA的bin文件夹里
在这个文件夹下找到相同名字的dll文件，直接复制进过去，再验证gpu是否可用，False变成True 在这里插入图片描述参考博客https://blog.csdn.net/a1456123a/article/details/115098630