环境搭建
安装显卡驱动
当前所用显卡为NVIDIA的P6000,在英伟达的官网上查看对应的驱动, 下载NVIDIA-Linux-x86_64-535.104.05.run并安装。
sudo ./NVIDIA-Linux-x86_64-535.104.05.run
安装成功后用nvidia-smi命令后查看
安装的cuda版本不能超过12.2,选择安装cuda11.8。
安装cuda
在英伟达官网上下载cuda 11.8并安装
sudo ./cuda_11.8.0_520.61.05_linux.run
安装结构报错:
查看log文件如下: