安装cuda11.0以及对应的cudnn后,发现最新的tensorflow-gpu=2.3.1仅支持cuda10.1。在运行gpu_test.py文件时出现了报错。
于是根据报错信息,执行命令
(base) lin@lin-desktop:/usr/local/cuda-11.0/lib64$ sudo cp ./libcudart.so.11.0 ./libcudart.so.10.1
(base) lin@lin-desktop:/usr/local/cuda-11.0/lib64$ sudo cp ./libcublas.so.11 ./libcublas.so.10
(base) lin@lin-desktop:/usr/local/cuda-11.0/lib64$ sudo cp ./libcusparse.so.11 ./libcusparse.so.10
(base) lin@lin-desktop:/usr/local/cuda-11.0/lib64$ sudo cp ./libcudnn.so.8 ./libcudnn.so.7
将cuda11.0的库复制并重命名为需要的库
再次运行gpu_test.py
输出
True
[’/device:GPU:0’]