报错信息: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中添加环境变量:
添加如下变量:
name:LD_LIBRARY_PATH
values: :/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64
然后再运行就不会报错了