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

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

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值