已知复现程序所需环境tf=1.12 cuda=9.0,而实验室服务器cuda 11.0,故使用anaconda创建虚拟环境,创建虚拟环境过程省略。
虚拟环境使用python3.6
配置代码如下:
pip install tensorflow-gpu==1.12
conda install cudatoolkit=9.0 # 这里本来想下9.2,奈何试了一下,不兼容
conda install cudnn # 自动下载对应的7.6.5版本
此处使用pip下载tensorflow1.12实属无奈,conda下载的同一个版本tensorflow会报错,无法运行所用程序的demo。
cuda下载cuda和cudnn的方法引用自虚拟环境下下载cuda,这是我能找到最方便快速的下载方法了。
补充说明
由于cuda9.0存在一些问题,我在进行数据预处理的时候弹了奇怪的bug,故通过cuda官方网站下载