anaconda安装tensorflow-gpu

今天用anaconda安装了tensorflow-gpu，可谓是耗尽心血。废话不多说，开始安装。

第一步 --- 安装anaconda和显卡驱动

超详细Anaconda安装教程_极小白的博客-CSDN博客

这个教程太多可以借鉴网上的文章或视频

第二步 --- 创建tensorflow-gpu虚拟环境

以管理员身份运行Anaconda Powershell Prompt

创建tensorflow-gpu虚拟环境

conda create -n tensorflow_gpu python=3.9

激活tensorflow-gpu虚拟环境

 

 

第三步 --- 安装tensorflow-gpu

 pip install tensorflow-gpu==2.6.0

这个文件大概423Mb ，然后开始等待

第四步 --- 遇到的问题

import tensorflow as tf
print(tf.test.is_built_with_cuda())
#Ture

print(tf.config.list_physical_devices('GPU'))
#[]

print(tf.test.is_gpu_available())
#False

 

Could not load dynamic library ‘cudnn64_8.dll‘；

dlerror: cudnn64_8.dll not found

这个问题会导致tensorflow无法调用GPU

我们在此电脑中搜索cudnn64_8.dll

 

 

找到这个文件，将这个文件复制到CUDA驱动文件夹下的bin文件夹下

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7\bin

---------------------------------------------------------------------------------------------------------------------------------

这样tensorflow-gpu的安装就结束了