安装tensorflow2及AttributeError问题解决

本文介绍了安装TensorFlow2.x时的注意事项,包括对Python、NVIDIA驱动、CUDA和cuDNN版本的要求。还提到了在Anaconda环境中安装的步骤,以及遇到的AttributeError问题,如找不到'cudart64_101.dll'和'tensorflow'模块无属性错误的解决方案。建议检查文件命名冲突并确保所有依赖项正确安装。
摘要由CSDN通过智能技术生成

初学tensorflow2.X,谈谈安装时踩的坑.

安装tensorflow2

1.方法一:tensorflow版本升到2.x后,直接终端pip install tensorflow已经是默认支持gpu版,中文官网说明https://tensorflow.google.cn/install/gpu?hl=zh_cn

  • 特别注意版本对应:tensorflow:2.1,python≥3.5,NVIDIA驱动版本≥418.X,NVIDIA CUDA版本≥10.1,否则后面有各种奇怪问题。
    NVIDIA® GPU 驱动程序:CUDA 10.1 需要 418.x 或更高版本。
    CUDA® 工具包:TensorFlow 支持 CUDA 10.1(TensorFlow 2.1.0 及更高版本)
    cuDNN SDK(7.6 及更高版本)

装完将 CUDA、CUPTI 和 cuDNN 安装目录添加到 %PATH% 环境变量中。例如,如果 CUDA 工具包安装到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1,同时 cuDNN 安装到 C:\tools\cuda,请更新 %PATH% 以匹配路径:

    SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin;%PATH%
    SET PATH=C:\Program Files\NVIDIA GPU Computing Toolki
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值