安装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

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。