一、Nvidia显卡驱动
一、安装Nvidia驱动
ubuntu-drivers devices
# 出现一堆信息,model后面是显卡型号,driver后面带有recommended的是推荐驱动安装版本
sudo ubuntu-drivers autoinstall
# 直接安装推荐版本
二、安装cuda
1. 直接上官网选择环境,并选择network安装方式,官方会自动适配合适的并且最新的安装指令,然后无脑执行。
官网:CUDA Toolkit 11.3 Update 1 Downloads | NVIDIA Developer
2. 改环境变量
vim ~/.bashrc
把下列信息(cuda版本看自己的)加在文件末尾,保存并退出
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.3/lib64
export PATH=$PATH:/usr/local/cuda-11.3/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.3
3. 重启
输入nvcc -V查看是否安装成功
三、cuDNN安装
1. 官网下载,选择cuDNN Library for Linux(x86,匹配的cuda版本)
cuDNN Archive | NVIDIA Developer
2. 在终端所cd的文件夹解压缩
tar -xzvf *********(cudnn文件名.tgz)
3. 执行下列命令
sudo cp cuda/include/* /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 /usr/local/cuda-11.3/lib64/libcudnn*
4. 用下列命令查看cuDNN版本
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
四、安装anconda
1. 下载Anaconda 免费版Anaconda | Individual Edition
2. 执行安装命令,中途均选确定
bash Anaconda3-2021.05-Linux-x86_64.sh
3. 更新环境变量
vim ~/.bashrc
在文件末尾加上 export PATH="/home/zcl(你的用户名)/anaconda3/bin:$PATH" 保存并退出
source ~/.bashrc