在服务器上import tensorflow 没问题,在pycharm中报错

报错信息:ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

代码在服务器上直接用命令行的方式运行没有问题,一旦在本地电脑上用pycharm连接,再跑的时候就报错。报错信息如上。

找了很久,有的说是cuda版本问题,有的说是环境变量问题,但是如果是cuda版本问题的话,服务器上命令行运行肯定也会报错的。

关于环境配置问题,之前也远程连接过服务器的环境,一直没有问题,这次出问题,我觉得主要是这次连的是服务器的gpu环境,所以需要额外配置一些环境变量(这些在以前都没有配置过)

配置的环境变量包括:

①检查一下自己服务器中的~/.bashrc文件,添加以下环境变量:

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64
export PATH=$PATH:/usr/local/cuda-9.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0
②在服务器的run-edit configurations中添加环境变量:

在服务器上import tensorflow 没问题,在pycharm中报错_第1张图片

添加如下变量:

name:LD_LIBRARY_PATH

values:   :/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64 

然后再运行就不会报错了

你可能感兴趣的:(在服务器上import tensorflow 没问题,在pycharm中报错)