今天运行tensorflow的时候,发现在pycharm下,程序无法找到CUDA的libcupti.so文件。而在添加完环境变量:
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH/usr/local/cuda/extras/CUPTI/lib64:
后,在命令行可以运行程序。然而,在Pycharm中运行程序,仍无法找到CUDA库文件。
解决方案:
在菜单Run->Edit configurations 中,手动设置Environment variables,添加LD_LIBRARY_PATH的内容:
LD_LIBRARY_PATH/usr/local/cuda/lib64
,重启pycharm即可解决问题。