tensorflow 安装cudnn和cudatoolkit时出现问题

hrirmatl  于 2022-12-13  发布在  其他
关注(0)|答案(1)|浏览(205)

我正在尝试使用tensorflow与GPU。我目前的cuda版本如下:

nvcc -V
    
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Tue_Mar__8_18:36:24_Pacific_Standard_Time_2022
Cuda compilation tools, release 11.6, V11.6.124
Build cuda_11.6.r11.6/compiler.31057947_0

当我尝试使用下列程式进行安装时:

conda install cudnn=8.7.0=cuda11.6_0

我收到以下错误:

PackagesNotFoundError: The following packages are not available from current channels:

  - cudnn==8.7.0=cuda11.6_0

我试着在谷歌上搜索以找到答案,但找不到任何有效的东西。一个解决方案声明使用!conda search cudnn来列出兼容的版本,但当我这样做时,我没有看到我安装的cuda(11. 6)版本...我收到了以下列表:

Loading channels: ...working... done
# Name                       Version           Build  Channel             
cudnn                          7.1.4       cuda8.0_0  pkgs/main           
cudnn                          7.1.4       cuda9.0_0  pkgs/main           
cudnn                          7.3.1      cuda10.0_0  pkgs/main           
cudnn                          7.3.1       cuda9.0_0  pkgs/main           
cudnn                          7.6.0      cuda10.0_0  pkgs/main           
cudnn                          7.6.0      cuda10.1_0  pkgs/main           
cudnn                          7.6.0       cuda9.0_0  pkgs/main      
cudnn                          7.6.4      cuda10.0_0  pkgs/main           
cudnn                          7.6.4      cuda10.1_0  pkgs/main           
cudnn                          7.6.4       cuda9.0_0  pkgs/main           
cudnn                          7.6.5      cuda10.0_0  pkgs/main           
cudnn                          7.6.5      cuda10.1_0  pkgs/main           
cudnn                          7.6.5      cuda10.2_0  pkgs/main           
cudnn                          7.6.5       cuda9.0_0  pkgs/main           
cudnn                          7.6.5       cuda9.2_0  pkgs/main           
cudnn                          8.2.1      cuda11.3_0  pkgs/main

有人能解决这个问题吗?

g6baxovj

g6baxovj1#

你不能通过conda安装cuDNN 8.7.0,因为你可以通过!conda search cudnn检查!
直接从NVIDIA网站下载。然后按照installation guide安装库。

相关问题