目录
一、安装显卡驱动
1.下载驱动:
官网连接:https://www.nvidia.cn/geforce/drivers/
前往官网选择合适驱动,比如我的设备是RTX 3070,用在64位的Linux上,搜索下载最新驱动即可。
2.删除现有驱动并安装相关依赖
为了确保驱动安装正确,先删除掉本机现有的驱动:
sudo apt-get remove --purge nvidia*
之后更行一下,并且安装一些相关的依赖:
sudo apt-get update
sudo apt-get install dkms build-essential linux-headers-generic
3.禁用nouveau(如果不禁用可能安装会不成功)
打开禁用配置文件:
sudo gedit /etc/modprobe.d/blacklist.conf
在配置文件的最下端加入以下禁用项:
blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lbm-nouveau off
禁用nouveau内核模块:
echo options nouveau modeset=0
sudo update-initramfs -u
重启设备:
sudo reboot
查看是否禁用nouveau成功,如果输入以下没有打印出任何内容,代表禁用成功:
lsmod | grep nouveau
4.安装显卡驱动:
关闭图形界面:
sudo systemctl set-default multi-user.target
重启计算机:
sudo reboot
进入非图形化界面,输入用户名密码并开始安装显卡驱动
sudo ./你下载的驱动
在安装过程中全程“yes”或者“continue”
安装完成开启图形化界面:
sudo systemctl set-default graphical.target
重启设备,设备图形化界面变正常:
sudo reboot
查看显卡是否安装成功并且查看该显卡适配CUDA版本号:
nvidia-smi
会出现显卡信息,代表安装成功,并从中可以看到CUDA版本号:
至此,显卡驱动安装完成
二、安装CUDA
CUDA下载地址:https://developer.nvidia.com/cuda-toolkit-archive
1.补充依赖库,并从CUDA官网下载合适的CUDA版本
比如我适用的是11.3,下载11.3版本即可:
补充依赖库:
sudo apt-get install freeglut3-dev libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev
2.检查下载的CUDA是否可用
在下载CUDA的路径下输入:
md5sum 你刚刚下载的CUDA安装包
出现类似这样的结果代表可用:
3.关闭图形化界面:
sudo systemctl set-default multi-user.target
sudo reboot
4.开启图形化界面:
sudo systemctl set-default graphical.target
sudo reboot
5.安装CUDA:
sudo sh 你刚刚下载的安装包
6.开始安装
(安装过程不需要再安装一次Nvidia驱动,因为刚刚我们已经安装过了)
安装完成后会在Summary界面看到ToolKit和Sample显示安装成功,Driver显示没有安装成功(因为我们刚刚没有安装显卡驱动)
7.添加环境变量:
sudo gedit ~/.bashrc
在最后加入:
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
保存并source一下:
source ~/.bashrc
8.检测是否安装成功
cat /proc/driver/nvidia/version
nvcc -V
输入以上指令会输出你安装的相应的CUDA版本
至此,CUDA安装完成
三、安装CUDNN
CUDNN下载官网:https://developer.nvidia.com/zh-cn/cudnn
1.下载CUDNN
需要登录Nvidia账户(当时登录好多次登录不上,多登录几次就好了),选择合适的CUDNN版本下载
2.解压CUDNN压缩包
可以右键单机压缩包解压后者使用命令解压
3.复制相应内容到CUDA下
sudo cp cuda/include/cudnn.h /usr/local/cuda-11.3/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.3/lib64
sudo chmod a+r /usr/local/cuda-11.3/include/cudnn.h
sudo chmod a+r /usr/local/cuda-11.3/lib64/libcudnn*
4.查看CUDNN版本:
cat /usr/local/cuda-11.3/include/cudnn.h | grep CUDNN_MAJOR -A 2
至此,CUDNN安装完成
四、安装anaconda
anaconda下载地址:
https://mirror.tuna.tsinghua.edu.cn/anaconda/archive/(国内镜像地址,下载比较快)
https://www.anaconda.com/products/individual(官网)
1.下载anaconda
从上述给出的网站选择你需要的anaconda版本下载
2.安装
sudo bash 你刚刚下载的文件地址
一路“yes”,其中可能会让你安装VSCode,自己可以选择以下要不要安装
3.环境配置
sudo gedit /etc/profile
在末尾加上
export PATH=/你的anaconda安装路径/bin:$PATH
4.查看是否配置成功
conda list
至此,anaconda安装成功