安装tensorflow-gpu踩坑总结

安装tensorflow-gpu踩得坑总结

  • 安装cuda之前要先安装合适的驱动
  • 下载cuda
  • 下载cuDnn
  • 下载anaconda

安装cuda之前要先安装合适的驱动

英伟达驱动下载地址

1.查看电脑的显卡类型
查看显卡类型的方法
2.选择对应的显卡类型和操作系统类型,下载驱动
3.下载好驱动以后按照如下网址查看支持的cuda版本
查看支持的cuda版本

下载cuda

百度搜索需要的cuda版本,进入官网下载对应版本
有local版和network版
尝试网络版失败,选择下载local版

下载cuDnn

下载地址
需要先注册登录才能有下载口出现
选择对应cuda版本的cuDnn就行了
下载之后把解压出来的三个文件夹丢进C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2
这个目录下面(因为我的是10.2,你的可能不是)

下载anaconda

去官网下载就好了,安装别忘了勾选环境变量
详细的步骤:
https://blog.csdn.net/zqx951102/article/details/89105110
下载前一定要看好版本
https://www.cnblogs.com/csu-lmw/p/10281607.html
cuda10往上tensorflow-gpu ==2.0.0是可以的
建议pip的时候把版本写清楚,不要按照默认版本。撞对版本的可能性太小。还得卸了再装。
使用pip下载太慢了可以换源,本来要下7个小时,结果几分钟就下完了。
https://blog.csdn.net/qq_40523737/article/details/83112522
尝试的时候可能会报错大概是找不到cudart64_100.dll
链接:https://pan.baidu.com/s/15LgSgKoxyDs9DPQg3kYeFA
提取码:0b4a
我找到的这个文件,把他丢进C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin
貌似问题就解决啦。

你可能感兴趣的:(安装tensorflow-gpu踩坑总结)