conda安装旧版本的tensorflow
前言:
可能有朋友需要复现一些比较老旧的论文需要用到落后版本的tensorflow-gpu,但是conda官方的tensorflow好像没有这么旧的,而且如果安装tensorflow-gpu,conda只会帮你安装cudnn,不会帮你安装cuda。如果用pip3安装,又会缺少对应的cuda和cudnn。
遇到问题:
程序-tensorflow版本-cuda-cudnn这四个版本不对应的话,就会出现网上很多类似的报错情况:
- Loaded runtime CuDNN library: 7600 (compatibility version 7600) but
source was compiled with 7102 (c - Loaded runtime CuDNN library: 7500 (compatibility version 7500) but
source was compiled with 7004
(诸如此类)
方法:
在conda库的搜索栏搜索自己tensorflow需要的版本,以及配对的cuda,cudnn版本。
tensorflow的cuda对应表:
据提示中所说,CUDA9和CuDnn7所对应的Tensorflow-gpu最高版本为1.12.0,后面的1.13.0均为CUDA10。
具体的cudnn版本我没有找到对应表,只能等出现问题的时候才知道程序所需要的cudnn版本是多少了,如7500 (compatibility version 7500) but source was compiled with 7004
,所需要的就是cudnn7.0。
以我此次为例,我需要的是tensorflow-gpu1.4.0,它对应的cuda8.0,cudnn==7.1,所以我就以这个为目标进行搜索安装,安装代码如下:
# cuda8.0
conda install -c mwojcikowski cudatoolkit
# cudnn7.1
conda install -c jjh_cio_testing/label/in_defaults cudnn
# tensorflow-gpu-base==1.4.0
conda install -c shuangnan tensorflow-gpu-base