ubuntu配置cuda环境

参考：https://blog.csdn.net/BigData_Mining/article/details/99670642

 

1. 安装合适的显卡驱动

 

2. 根据英伟达官网给出的命令安装cuda

 

3. 下载相应版本的cudnn包，放到cuda安装路径中相应位置

 

4. 配置环境

修改/home/.bashrc的最后添加如下两行：

export PATH="/usr/local/cuda-10.0/bin:$PATH"
export LD_LIBRARY_PATH="/usr/lcoal/cuda-10.0/lib64:$LD_LIBRARY_PATH"

让配置立刻生效：

source .bashrc

 

报错：ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory，检查版本对应无问题

在命令行中输入：sudo ldconfig /usr/local/cuda-10.0/lib64

输出为：/sbin/ldconfig.real: /usr/local/cuda-10.0/lib64/libcudnn.so.7 is not a symbolic link

解决办法：sudo ln -sf /usr/local/cuda/lib64/libcudnn.so.7.4.1 /usr/local/cuda/lib64/libcudnn.so.7

（建立软链接，其中libcudnn.so.7.4.1为cudnn中的文件）