Tensorflow+Cudnn配置

首先检查自己的cuda版本

!nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Wed_Feb__8_05:53:42_Coordinated_Universal_Time_2023
Cuda compilation tools, release 12.1, V12.1.66
Build cuda_12.1.r12.1/compiler.32415258_0

然后下载对应Cudnn
https://developer.nvidia.com/rdp/cudnn-download

下载完毕后将文件拷到X:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\对应目录下,拷完以后不需要重启系统也不需要重启Pycharm就可以正常使用。

你可能感兴趣的:(机器学习,tensorflow,人工智能,python)