安装tensorflow2.4-gpu方法

安装tensorflow2.4-gpu方法

1.打开Anaconda Prompt
2.命令行输入:conda create -n TF2.4 python==3.6
提示后输入y并回车安装相关附带的包
3.命令行输入:conda activate TF2.4
4.命令行输入:命令行输入:conda install cudatoolkit=11.1
5.命令行输入:conda install cudnn=8.1
6.命令行输入:pip install tensorflow-gpu==2.4
7.还需要安装对应版本的keras2.4.3
在这里插入图片描述
由于一些版本可能找不到可以用以下指令添加镜像源
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --set show_channel_urls yes
注明:30系显卡需要cudnn大于8.05,cudatoolkit不小于11.1

Logo

分享最新的 NVIDIA AI Software 资源以及活动/会议信息,精选收录AI相关技术内容,欢迎大家加入社区并参与讨论。

更多推荐