1.Ubuntu上安装nvidia driver
- 在https://www.nvidia.com/Download/index.aspx上根据系统版本和显卡的型号选择适合的驱动进行下载
- 下载驱动软件完成后,如我下载的是NVIDIA-linux-x86_64-470.94.run,将其设置为可执行文件:
sudo chmod +x NVIDIA-linux-x86_64-470.94.run
- 按下CTRL + ALT + F1进行命令行界面
- 卸载本机上的nvidia driver:
sudo nvidia-uninstall
,如果没有可以跳过 - 删除相关文件:
sudo apt purge nvidia*
- 安装驱动:
sudo ./NVIDIA-linux-x86_64-470.94.run
注:如果在安装过程中出现报错:ERROR: You appear to be running an X server; please exit X before installing
,可以执行指令sudo service lightdm stop
进行关闭,然后重新执行上述过程。
2.Ubuntu上安装cuda toolkit
如第一步所示,我已经装好了470.94版本的nvidia driver,最高可以支持CUDA Version:11.4,但是我有使用cuda10.2的需求,那么我就去安装对应版本的cuda,在https://developer.nvidia.cn/cuda-toolkit-archive可以找到各版本的cuda toolkit,选择需要的然后下载即可。
我下载了cuda_10.2.89_440.33.01_linux.run,然后sudo sh cuda_10.2.89_440.33.01_linux.run
,输入accept,只安装+号对应cuda toolkit进行安装,其他的不用选,否则容易报错。
在安装完成之后,就会发现/usr/local/cuda-10.2目录已经存在了。