ubutnu18.04、cuda10.2安装tensorflow-gpu==2.3.0、keras==2.4.3
安装指令:
pip install tensorflow-gpu==2.3.0 -i https://mirror.baidu.com/pypi/simple
安装完毕tensorflow=2.3.后,对能否调用gpu进行校验:
执行指令:
import tensorflow as tf
print(tf.test.is_gpu_available())
返回为true,则表示安装成功,如果False表示安装失败,错误提示信息如下:
Could not load dynamic library 'libcudart.so.10.1'; dlerror: libcudart.so.10.1: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /usr/local/cuda/lib64
2022-05-10 15:40:08.200478: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your ...
解决方案:
conda install cudatoolkit=10.1