本教程主要是参考以下大神的帖子,这里先表示感谢。
https://blog.csdn.net/weixin_38106878/article/detail/88081923?spm=1001.2014.3001.5501
一. 安装NVIDIA显卡
安装步骤:
- 添加源
sudo add-apt-repository ppa:graphics-drivers/ppa
- 更新源
sudo apt-get update
- 寻找合适的NVIDIA驱动
ubuntu-drivers devices
- 安装驱动(把后面的数字改成合适的版本号即可)
sudo apt-get install nvidia-450
- 重启Ubuntu系统,并验证是否安装成功,若显示版本号,安装成功。
nvidia-smi
二. 安装Anaconda管理环境
- 在anaconda官网下载Linux版本的下载包
- 在终端中运行
cd Downloads/
sudo sh Anaconda3-2020.11-Linux-x86_64.sh
- 进入注册信息页面,输入yes
- 阅读注册信息,然后输入yes;查看文件即将安装的位置,按enter,即可安装.
- 安装完成后,收到加入环境变量的提示信息,输入yes.
- 提示信息“Do you wish to proceed with the installation of Microsoft VSCode? [yes|no]”,输入no.
- 重启终端,即可使用Anaconda3。
三. 安装CUDA
- 去NVIDIA官网下载相应cuda。
- 将下载好的文件放置在对应的目录下,并输入指令执行。
sudo sh cuda_8.0.27_linux.run
- 开始安装,按空格阅读安装证书内容,并确认安装。在提示到安装显卡时,我们不再安装显卡,因为之前我们已经安装好。 其他的安装选项用使用默认安装。
- 安装完成后声明环境变量,并将其写入到 ~/.bashrc 的尾部:
sudo gedit ~/.bashrc
// export PATH=/usr/local/cuda-8.0/bin$ {PATH:+:$ {PATH}}
//export LD_LIBRARY_PATH=/usr/local/cuda-8.0 /lib64$ {LD_LIBRARY_PATH:+:$ {LD_LIBRARY_PATH}}
- 然后设置环境变量和动态链接库,在命令行输入。
sudo gedit /etc/profile
- 在打开的文件末尾加入>
export PATH=/usr/local/cuda/bin:$PATH
- 测试cuda是否安装成功>
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
make
sudo ./deviceQuery
- 如果终端显示出显卡信息,则表示安装成功。
四. 安装CUDA
- 去nvidia官网下载对应版本的cudnn,这里我下载的是cudnn v5.1。下载下来后解压到自己指定的目录下;
- 进入目录,建立软件链接(注意自己cudnn版本号):
sudo cp ~/cuda/include/cudnn.h /usr/local/cuda-8.0/include
sudo cp ~/cuda/lib64/lib* /usr/local/cuda-8.0/lib64
cd /usr/local/cuda/lib64
sudo rm -rf libcudnn.so libcudnn.so.5 #删除原有动态文件
sudo ln -s libcudnn.so.5.0.5 libcudnn.so.5 #生成软衔接
sudo ln -s libcudnn.so.5 libcudnn.so
sudo ldconfig