Ubuntu 20.04下运行TensorFlow GPU的Tips

1 About CUDA

1.1 Environment

参照官方文档和“CUDA安装成功后配置环境变量”，在~/.bashrc和/etc/profile中配置好path-to-cuda-bin和LD_LIBRARY_PATH即可。

可能设置好和环境变量后运行TensorFlow时，仍然会报Could not dlopen library 'libcudart.so.10.0'等错误，此时参照文档，运行下列命令即可，其中x 处为对应版本号。

sudo ldconfig /usr/local/cuda-10.x/lib64

2 About cuDNN

2.1 Environment

参照官方文档，下载与CUDA版本对应的cuDNN，下载后运行dpkg -i xxx.deb来安装。

可能安装cuDNN后，运行TensorFlow会出现报Could not dlopen library 'libcudart.so.10.0'等错误，此时只需按照1.1，重新运行下列命令即可。

sudo ldconfig /usr/local/cuda-10.x/lib64