安装tensorflow / cuda/ cudnn

1. win10下查看cuda版本 

到桌面点击鼠标右键,选择NVIDIA控制面板, 在Nvidia面板中点击帮助,并选择系统信息, 然后选择选择组件,找到“NVCUDA.DLL”,即可看到支持的CUDA版本

如下图, CUDA版本号: 10.1.120


或者在命令提示符下敲入命令 : nvidia-smi 

2. cuDnn的版本下载: https://developer.nvidia.com/rdp/cudnn-archive  

3. CUDA的版本下载: https://developer.nvidia.com/cuda-toolkit-archive 

4. tensorflow/CUDA/cuDnn 版本的对应关系:https://github.com/fo40225/tensorflow-windows-wheel


碰到的问题:

1. Could not load dynamic library ‘cudart64_100.dll’; dlerror: cudart64_100.dll not found

cudart64_100.dll 这个文件需要从cuda10的版本copy来

2. InternalError: Blas GEMV launch failed

cublas64_100.dll(10改100)

参考:

https://blog.csdn.net/weixin_43026262/article/details/95742529 

https://blog.csdn.net/XunCiy/article/details/89069562

你可能感兴趣的:(安装tensorflow / cuda/ cudnn)