您好,欢迎来到五一七教育网。
搜索
您的当前位置:首页3090显卡TensorFlow,CUDA,cuDNN 深度学习环境配置

3090显卡TensorFlow,CUDA,cuDNN 深度学习环境配置

来源:五一七教育网

1.找到对的版本

      在官网找到想要的TensorFlow版本,及相对应的cuda,cuDNN的版本

网址:

因为3090深度学习需要cuda版本在11.0以上 所选择框选的版本。

 

 顺带一提 cuda版本的选择也需要显卡驱动的兼容 可以Win+r搜索 cmd 输入:

nvidia-smi

结果如下   框选处表示此驱动支持的最高cuda版本。

 2.cuda11.0的安装

我选择的版本是2.4 在Anaconda创建的虚拟环境中安装

1.打开anaconda  输入:(enc_name 是你自己对这个环境的命名)

conda create -n env_name python=3.7

 选择Y回车

 运行完后进入刚刚创建的环境(enc_name 是你自己对这个环境的命名)

conda activate env_name

安装cuda

conda install cudatookit=11.0

然后选择y确认安装

3.cuDNN8.0安装

输入如下命令,片刻后选择y确认安装

conda install cudnn==8.0.5.39 -c conda-forge

4.TensorFlow安装

输入如下命令,片刻后选择y确认安装

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple/ --upgrade tensorflow==2.4

 安装完成后如下操作 输出2.4.0即表示安装成功。

因篇幅问题不能全部显示,请点此查看更多更全内容

Copyright © 2019- 517ttc.cn 版权所有 赣ICP备2024042791号-8

违法及侵权请联系:TEL:199 18 7713 E-MAIL:2724546146@qq.com

本站由北京市万商天勤律师事务所王兴未律师提供法律服务