检查自己的GPU是否支持CUDA,linux可用以下命令来查看GPU型号:
lspci | grep –i nvidia
本机显示结果如下
03:00.0 3D controller: NVIDIA Corporation GK110BGL [Tesla K40c] (rev a1)
04:00.0 VGA compatible controller: NVIDIA Corporation GK110GL [Quadro K5200] (rev a1)
04:00.1 Audio device: NVIDIA Corporation GK110 HDMI Audio (rev a1)
只要GPU型号可以在https://developer.nvidia.com/cuda-gpus找到,就是支持CUDA的,同时还可以查看显卡的计算能力,数值越大越好。
检查自己的操作系统是不是CUDA支持的,可以输入以下命令:
uname -m && cat /etc/