一、Nvidia Tesla P40显卡驱动安装

1、确定显卡硬件是否有问题

1)通过终端命令确定显卡是否插好

sudo lshw -C display

如果输出有独显信息则证明显卡已插好

2)确定显卡电源是否有连接好

2、安装显卡驱动

一种方式是通过cuda一键安装,另一种是通过apt安装

sudo apt install nvidia-driver-560

其中560换成自己需要的显卡驱动版本,可以通过cuda官网查看最新的cuda版本所对应的显卡驱动版本,然后再通过nvidia官网查看是否有对应型号的驱动如果有则直接执行上述命令,如果没有,则选择降cuda的版本,直至找到对应的

注意:如果之前系统中有安装过显卡驱动但是想安装新的,那么一定要将之前所安装的驱动删除干净,具体做法如下

sudo apt-get purge nvidia*
sudo apt-get autoremove

之后查看是否有残余,如果有则强制删除

dpkg -l | grep nvidia
sudo dpkg --purge --force-all <待删除文件名>

通过下述命令检查是否安装成功

nvidia-smi

二、安装Cuda

1、上cuda官网CUDA Toolkit 12.6 Update 3 Downloads | NVIDIA Developer找到能够与显卡所支持的最新版本驱动对应的.run文件,该官方网站会提供安装终端命令

wget https://developer.download.nvidia.com/compute/cuda/12.6.3/local_installers/cuda_12.6.3_560.35.05_linux.run
sudo sh cuda_12.6.3_560.35.05_linux.run

如果已经安装了驱动那么在安装时把驱动选项默认的对号弄掉,之后继续安装,

三、安装Pytorch

Start Locally | PyTorch这个网址根据已安装的Cuda版本安装pytorch版本,注意安装的pytorch版本所支持的Cuda版本一定要低于或等于所安装的Cuda版本,具体安装命令该官方网站会提供

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐