Ubuntu安装cuda及相关问题

安装cuda

本教程建议从Nvidia官网下载run离线包进行安装。

首先禁用Ubuntu自带的nouveau驱动:

新建文件 /etc/modprobe.d/blacklist-nvidia-nouveau.conf,写入以下内容:

blacklist nouveau
options nouveau modeset=0

更新 kernel initramfs:sudo update-initramfs -u,最后重启。

安装必要的软件:sudo apt install gcc make,最后使用命令sudo sh cuda_*.run安装cuda。

安装完成后注意命令行中提示的注意事项。

安装cudnn

tar -zxvf cudnn-*.tgz  sudo cp cuda/lib64/* /usr/local/cuda/lib64  sudo cp cuda/include/* /usr/local/cuda/include/

Add these contents to ~/.bashrc

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

then run

$ source ~/.bashrc