今天运行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的内容,即可解决问题。