1.找到对的版本
在官网找到想要的TensorFlow版本,及相对应的cuda,cuDNN的版本
网址:在 Windows 环境中从源代码构建 | TensorFlow (google.cn)
因为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即表示安装成功。