环境配置:V100服务器配置Tensorflow1.13+Keras2.2.4

V100服务器配置Tensorflow1.13+Keras2.2.4

  • Tensorflow与Keras版本匹配
  • Tensorflow与CUDA版本匹配
  • conda创建和配置环境
    • 创建虚拟环境
    • 安装Tensorflow对应的cuDNN和CUDA

Tensorflow与Keras版本匹配

环境配置:V100服务器配置Tensorflow1.13+Keras2.2.4_第1张图片
更多匹配信息参考

Tensorflow与CUDA版本匹配

环境配置:V100服务器配置Tensorflow1.13+Keras2.2.4_第2张图片

conda创建和配置环境

创建虚拟环境

conda create -n name python=3.6 numpy pandas

安装Tensorflow对应的cuDNN和CUDA

source activate name
conda install cudnn=7.4 cudatoolkit=10 # 报错,conda找不到cudnn=7.4,因此只安装cudatoolkit
conda install cudatoolkit=10 #成功

查看清华镜像网站,发现没有cudnn=7.4,因此改为安装cudnn=7.6

conda install cudnn=7.6 #成功
conda install tensorflow-gpu=1.13

啊我突然发现conda吧我的cuda自动改成和tensorflow对应的版本了,所以conda应该是可以自动安装cuda的,踩坑踩坑。
下次,无视cuda,直接conda install 需要的tensorflow版本即可。

conda install keras-gpu=2.2.4

你可能感兴趣的:(tensorflow,python,人工智能)