参考:https://blog.youkuaiyun.com/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中的文件)