安装显卡驱动:
详细过程可参照我的另一篇博客:
https://blog.csdn.net/qq_42683011/article/details/115436242
安装CUDA:
参考博客:
https://blog.csdn.net/xueshengke/article/details/78134991
到官网下载完合适版本的cuda .run
安装包后, 直接以root身份运行:
./cuda_11.2.0_460.27.04_linux.run
之后会解压, 并进入安装界面
由于我们之前已经安装完成了显卡驱动, 所以这里不要选择安装驱动
其他选项维持默认即可
之后会进入安装, 时间较长, 耐心等待即可
安装完成后, 会显示:
提示需要将两个环境变量添加到系统变量中
vim
进到/etc/profile
中, 在最后添加:
# CUDA PATh
export PATH=/usr/local/cuda-11.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.2/lib64:$LD_LIBRARY_PATH
source /etc/profile
添加完成后, 重启机器, 并再次进到/etc/profile
中注释掉最后一句
CUDA测试:
测试 cuda, nvcc 命令是否可用:
[root@localhost ~]# cuda 按两下tab
cudafe++ cuda-gdbserver cuda-memcheck
cuda-gdb cuda-install-samples-11.2.sh cuda-uninstaller
[root@localhost ~]# nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Mon_Nov_30_19:08:53_PST_2020
Cuda compilation tools, release 11.2, V11.2.67
Build cuda_11.2.r11.2/compiler.29373293_0
并测试CUDA例程:
找到cuda_sample目录, 我的是在cuda安装包的目录下:
进入到目录中, 编译运行几个常用的CUDA测试程序:
cd NVIDIA_CUDA-11.2_Samples/1_Utilities/
此目录下包含常用的测试程序:
[root@localhost 1_Utilities]# ls
bandwidthTest deviceQuery deviceQueryDrv p2pBandwidthLatencyTest topologyQuery UnifiedMemoryPerf
[root@localhost 1_Utilities]# cd deviceQuery
[root@localhost deviceQuery]# make
如果make出现问题:
安装一下g++就好:
yum install gcc-c++ -y
之后就能正常make并运行程序了:
另一个测试程序:
[root@localhost 1_Utilities]# cd bandwidthTest/
[root@localhost bandwidthTest]# make