Couldn‘t open CUDA library libcupti.so

在训练模型时使用print(tf.test.is_gpu_available())显示True,但是在训练时仍然会显示

Couldn't open CUDA library libcupti.so 。

是因为没有为libcupti添加环境变量，其实这一点在官方文档中已经指出了，再次证明查阅官方文档的重要性，不慌，只需要先

cd ~
vi .bashrc

之后添加

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda

然后

source .bashrc

重新导入环境变量即可
附上一张成功的截图