V100服务器配置Tensorflow1.13+Keras2.2.4
Tensorflow与Keras版本匹配
Tensorflow与CUDA版本匹配
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