炼丹初始化 - CentOS安装Nvidia显卡驱动+CUDA+cuDNN+PyTorch_centos安装cudnn

对于学习变音模型训练流程，必不可少的过程就是安装炼丹环境，而且对于新手而已，坑非常多，此处做学习记录，有遗漏或错误之处，望各位大大指出！

安装Nvidia显卡驱动

首先需要查看设备的显卡型号：

lspci | grep -i nvidia

输出结果如下：

此处着实不懂为啥没有显示型号，希望各位大神解惑一下（我的机器应该是GeForce GTX 3060）

先进入管理员权限：

sudo -s
# 然后输入密码并回车

开始安装Nvidia显卡驱动：

1、首先要禁用自带的Nouveau驱动，因为会和Nvidia驱动有冲突

# 编辑文件：
vim /etc/modprobe.d/blacklist-nouveau.conf
 
# 在文件最后加入以下内容
blacklist nouveau
options nouveau modeset=0
 
# 然后更新，并重启机器
sudo dracut --force
reboot
 
# 查看是否禁用成功，输入
lsmod |grep nouveau
 
# 若没有输出，则禁用成功
 

2、安装基础依赖环境

yum -y install gcc kernel-devel kernel-headers cmake

3、在Nvidia官网下载相应驱动

官网下载地址：https://www.nvidia.cn/Download/index.aspx?lang=cn

进入下载后的目录：

然后赋予脚本执行权限，执行脚本：

chmod +x NVIDIA-Linux-x86_64-535.146.02.run
sh NVIDIA-Linux-x86_64-535.146.02.run -no-opengl-files -no-nouveau-check

然后一直选择yes，回车即可安装成功！

检查是否安装成功：

nvidia-smi

出现下图所示即是安装成功：

安装Nvidia显卡驱动版本是535.86.10，CUDA版本是：12.2

注意：以下是本人查找资料时大佬给出的问题及解决方案，可以参考，但本人并未遇到

安装对应版本的CUDA（此处为12.2版本）

安装CUDA前确保以下工作已完成：

1、确认已成功安装 显卡驱动
2、nouveau已经禁用
3、系统已经安装了gcc

下载对应版本的CUDA，我下载的是12.2版本的

官网下载链接：https://developer.nvidia.cn/cuda-toolkit-archive

我下载的版本是：

然后给脚本赋予可执行权限：

chmod +x cuda_12.2.1_535.86.10_linux.run

首先需要输入accept：

然后使用默认选择的选项，直接Install

安装完后（若有警告不必理会）进行环境变量配置：

vim ~/.bashrc  
# 末尾添加
export CUDA_HOME=/usr/local/cuda-12.2    
export LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64:$LD_LIBRARY_PATH   
export PATH=/usr/local/cuda-12.2/bin:$PATH

终端输入：

nvcc --version

 显示结果如下，即安装成功：

安装cuDNN（需要对应Nvidia、cuda版本选择12.X）

官网下载链接：cuDNN Archive | NVIDIA Developer

选择对应版本下载（此处下载需要注册）：

下载完成会得到一个tgz压缩包：cudnn-12.2-linux-x64.tgz

进入下载的文件所在目录，解压：

tar -zxvf cudnn-12.2-linux-x64.tgz

解压后会得到一个cuda-12.2文件夹

需要将cuda-12.2/include/cudnn.h和cuda-12.2/lib64/libcudnn*拷贝到之前安装的cuda的文件夹下：

cp cuda-12.2/include/cudnn.h  /usr/local/cuda-12.2/include 
cp cuda-12.2/lib64/libcudnn*  /usr/local/cuda-12.2/lib64

至此，cuDNN安装完成，接下来就可以安装PyTorch去调用显卡了。

安装anaconda

参考博客：CentOS 7 安装Anaconda及简单配置_centos anaconda安装-CSDN博客

安装PyTorch

创建conda环境：

conda create --name <环境名称> python=需要的py版本
 
eg ： conda create --name test python=3.8

由于我的cuda版本是12.2，所以cu118及以下的都可兼容，此处安装的是torch2.1.0+cu118版本

pip install torch==2.1.0+cu118 torchvision==0.16.0+cu118 torchaudio==2.1.0+cu118 -f https://download.pytorch.org/whl/cu118/torch_stable.html

然后终端输入：

# 进入python代码编辑环境
python
 
# 输入
import torch
torch.__version__
print(torch.cuda.is_available())

输出下图所示即安装成功：

 

至此炼丹条件已具备，即可开启炼丹过程！！！

希望各位大神指出问题啊，本人是个刚毕业小白，欢迎高人指点！

（本人是非常热爱学习的！！！）