【注意!!!】
cuda
和cuDNN
的版本必须和tensorflow
的版本对应,否则会提示找不到对应的.dll
文件,tensorflow
官网提示如下:
查看Tensorflow-gpu对应的cuda, cuDNN 版本:
在创建的虚拟环境中通过命令安装
conda create -n test tensorflow==2.3 cudatoolkit=10.1 cudnn=7.6
【注意!!!】
cuda
和cuDNN
的版本必须和tensorflow
的版本对应,否则会提示找不到对应的.dll
文件,tensorflow
官网提示如下:
查看Tensorflow-gpu对应的cuda, cuDNN 版本:
conda create -n test tensorflow==2.3 cudatoolkit=10.1 cudnn=7.6