第一步:配置显卡驱动
1.安装英伟达显卡的驱动版本(手动配置)
1.1确认你的显卡支持和兼容的CUDA版本
nvidia-smi
注:neuralplexer2要求CUDA>=10.2
1.2CUDA安装
官网链接:
CUDA Toolkit Archive | NVIDIA Developer
并不一定要安装最高的适配版本,因为后续要和PyTorch适配,我这里安装的就是12.1。
#官网指导的安装代码
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.2.1/local_installers/cuda-repo-ubuntu2004-12-2-local_12.2.1-535.86.10-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-12-2-local_12.2.1-535.86.10-1_amd64.d