写好的项目,在 Pycharm 里运行是没有这个问题的,
即便是 .bashrc
下面没有写入 cuda 的路径,
但是当直接用命令行跑程序的时候,这个问题又蹦出来了
error while loading shared libraries: libcudart.so.8.0: cannot open shared object file: No such file or directory
猜测,应该是 PyCharm 帮我们做了引入库的操作。
这样,就需要修改 .bashrc
,添加这三行:
export PATH=/usr/local/cuda/bin:$PATH
export LIBRARY_PATH=/usr/local/cuda/lib64:$LIBRARY_PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
如果仍然有问题,执行:
sudo cp /usr/local/cuda/lib64/libcudart.so.8.0 /usr/local/lib/libcudart.so.8.0 && sudo ldconfig
sudo cp /usr/local/cuda/lib64/libcublas.so.8.0 /usr/local/lib/libcublas.so.8.0 && sudo ldconfig
sudo cp /usr/local/cuda/lib64/libcurand.so.8.0 /usr/local/lib/libcurand.so.8.0 && sudo ldconfig