前些天不知为啥cuda不能用了,nvidia-smi
也没反应。然后我就重新装了一下cuda。后来使用pycharm远程连接时,居然报错了。
ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
搜了一下网上的解决方法,基本上都是临时的添加变量的方法,这太难受了。我测了一下pytorch
也是可以正常使用,就是tensorflow
不行。
最后在 https://github.com/tensorflow/tensorflow/issues/21074。找到了解决方法
sudo vim /etc/environment
PATH="/usr/local/cuda-9.0/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/games:/usr/local/games"
LD_LIBRARY_PATH="/usr/local/cuda-9.0/lib64"
这里添加的路径和自己的一样哈。