记录:在复现论文代码时碰到使用keras环境,于是在自己windows系统的台式机(RTX 3080;CUDA 11.7)上进行了安装,但是发现台式机的显存无法支持程序的运行。于是将一摸一样的环境配置到更大现存的Ubuntu服务器(CUDA 11.7)上,但配置环境出错,一直无法调用GPU。经过一天的摸索,从网上其他博客中找到答案,虽然不懂其中的道理,但先记录下来。
conda create -n tf-gpu python=3.8.0
conda activate tf-gpu
conda install cudatoolkit=11.3.1
conda install cudnn=8.2.1
pip install tensorflow-gpu==2.7.0
pip install tensorflow-gpu==2.7.0 -i https://pypi.tuna.tsinghua.edu.cn/simple --user
tf.test.is_gpu_available()
检查Tensorflow中是否可用GPU,出现“True”则可用,出现“Flase”则表示GPU不可用。pip install keras==2.7.0 -i https://pypi.tuna.tsinghua.edu.cn/simple --user
nvidia-smi
查看已有CUDA版本为11.7,待配置的环境版本与Windows配置的版本一致:tf.test.is_gpu_available()
检查是否能够使用GPU时出现“Flase”:Could not load dynamic library 'libcudnn.so.8'; dlerror: libcudnn.so.8: cannot open shared object file: No such file or directory
locate libcudnn.so.8
(找到该动态库)tf.test.is_gpu_available()
检查是否能够使用GPU(此时提示“True”证明GPU可用。若仍然提示“Flase”,可按照上述几行代码,将无法加载的动态库都找到对应的路径且写入环境文件中。我在配置环境中除了libcudnn.so.8还有libcudart.so.11.0,通过上述几行代码成功的解决了无法使用GPU的问题):pip install keras==2.7.0
安装keras,检查后确认成功安装!上述安装Tensorflow-gpu以及Keras过程中,我参考两个博客,传送:,