问题描述
在使用pycharm远程安装tensorflow时遇到问题:
运行测试脚本
import tensorflow as tf
print(tf.test.is_built_with_cuda())
print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))
报错:
可见测试出GPU可用数量为0,并报错Could not load dynamic library 'libcudnn.so.8'; dlerror: libcudnn.so.8: cannot open shared object file: No such file or directory检测不到这个动态链接。
查验
经检查此文件就在/usr/local/cuda-11.7/targets/x86_64-linux/lib/目录下且此地址LD_LIBRARY_PATH居于首位。经检验重装tensorflow版本依旧不行(我的配置为cuda11.7,cudnn82,tensorflow2.8.0)
使用MobaXterm终端运行测试脚本惊奇发现tensorflow没问题:
解决方案
并且在测试当中发现,若先导入torch在pycharm中运行测试脚本也能正常检测到GPU,不知道是什么原因导致了这个问题。
import torch
import tensorflow as tf
print(tf.test.is_built_with_cuda())
print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))
希望有大佬能够帮忙分析解决一下。