报错都是同一个,错误如下:
在ubuntu下面运行错误
Failed to find dynamic library: libnccl.so ( libnccl.so: cannot open shared object file: No such file or directory )
而在idea或者其它工具下报错
Failed to find dynamic library: libcublas.so ( libcublas.so: cannot open shared object file: No such file or directory )
如果已经正确安装了gpu的情况下错误(nvcc -V命令查看gpu是否可用),可参照下面2种方式:
1.通过终端,输入:sudo ldconfig /usr/local/cuda/lib64,之后运行程序看是否正常
1. 通过环境变量
在~/.bashrc 中的变量,$LD_LIBRARY_PATH放在/usr/local/cuda/lib64前面,这个路径中不要使用变量!不要使用变量!不要使用变量!
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
如果还是不行尝试下将
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64 替换成
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64