Linux服务器3090上安装keras环境(3步走) 您所在的位置:网站首页 服务器3090 Linux服务器3090上安装keras环境(3步走)

Linux服务器3090上安装keras环境(3步走)

2024-07-06 04:09| 来源: 网络整理| 查看: 265

1.首先Linux服务器中安装了anaconda; 2.检查服务器的cuda版本,因为要安装的python,tensorflow及tensorflow-gpu, keras与cuda版本都要对应; keras版本一般比tensorflow要高一点

说下个人的情况: RTX3090的卡 cuda是11.2 需要安装的tensorflow及tensorflow-gpu版本为2.6.0 因此python至少是3.7 keras我装的是2.6.0

所有用到的代码如下: conda create --name mykerasenv3.7 python=3.7 #创建新环境 conda activate mykerasenv3.7 #激活创建好的环境 conda install tensorflow==2.5.0 #安装tensorflow pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==2.5.0 安装tensorflow-gpu conda install keras #安装keras 1.创建python3.7的新环境 conda create --name mykerasenv3.7 python=3.7 #创建一个python3.7版本环境,命名为mykerasenv3.7 2.验证cuda 以及 cudnn(gpu版本,可选)

用nvidia-smi查看自己的cuda,我的是11.2

在这里插入图片描述

3.安装tensorflow及tensorflow-gpu conda activate mykerasenv3.7 #先激活创建好的环境,开始安装包 # 加的一些镜像,加快下载速度 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ conda config --set show_channel_urls yes conda install tensorflow==2.5.0 #安装tensorflow  conda install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==2.5.0  #安装tensorflow-gpu 4.安装keras conda install keras #安装keras

这里给我下载的是2.4.3的,而我的tensorflow是2.5.0,直接运行代码出错了,说什么cudnn的问题,实际上是keras版本的问题 于是我直接在pycharm中更新keras到2.6.0 在这里插入图片描述 运行成功!

但是发现用得是cpu在跑 更新keras到2.6.0后,tensorflow也跟着被跟新到2.6.0了,导致比tensorflow-gpu的版本高,用cpu跑,这里又更新了下tensoflow-gpu到2.6.0,与cuda版本是对应的 conda install tensorflow-gpu==2.6.0



【本文地址】

公司简介

联系我们

今日新闻

    推荐新闻

    专题文章
      CopyRight 2018-2019 实验室设备网 版权所有