Ubuntu 20.04下运行TensorFlow GPU的Tips

it2024-07-23  39

Tips List

1 About CUDA1.1 Environment 2 About cuDNN2.1 Environment

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
最新回复(0)