一、问题提出
之前写好的代码都是基于tensorflow 1.14.0的,在自己笔记本上跑太慢了,所以搬到服务器去跑。但是一个很现实的问题就是服务器的CUDA是10.2的,因为服务器有很多人在用,其他人用的是10.2的,而tensorflow 1.14.0对应的CUDA版本是10.0的,我也不太可能去卸载了去重新装一个CUDA,在tensorflow的安装过程中不会出现大的问题,但是在后面运行的时候会报错,会提示找不到:cudart64_100.dll文件,看名称就可以看得出来是64位10.0版本的CUDA的dll文件,装好的10.2只有cudart64_102.dll,当然找不到了,所以就尝试去寻找在CUDA10.2的基础上安装tensorflow-GPU 1.14.0,并且成功运行。
注:本文的前提是已经安装好CUDA和CUDNN,然后安装tensorflow-GPU 1.14.0
二、问题解决
1.在服务器上建立虚拟环境进行安装:
conda create --name panda python=3.6
其中panda是虚拟环境的名称,可以自行命名
2.激活环境
activate panda
3.查看环境
conda info --envs
只要下面显示你刚建立的虚拟环境即可
注:采用pip install tensorflow进行安装时,在之后的运行过程中要求会报错显示CUDA必须为10.0版本,而采用conda方式安装则可以避免这个问题,过程中发现