tensorflow2.4-gpu
·
安装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
更多推荐




所有评论(0)