1 设置环境变量,加入PACKAGES
sudo gedit /etc/profile
在下面添加cuda路径,输入
PACKAGES = /usr/local/cuda-10.1
保存后,使用指令
source /etc/profile
2 执行dspsr configure
cd dspsr/
./configure
3 确认CUDA是否被识别,使用指令
grep CUDA Makefile
如果出现
CUDA_CFLAGS = -I/usr/local/cuda-10.1/include
CUDA_LIBS = -L/usr/local/cuda-10.1/lib64 -lcudart
CUDA_NVCC = /usr/local/cuda-10.1 XXXXXXX
则证明configure成功,如果没有出现 检查两方面① 环境变量配置是否正确 ②NVIDIA驱动是否正确安装(nvidia-smi检查)
4 然后make,使用指令
make
sudo make install
5 打开dspsr help 检索cuda 使用指令
dspsr -h | grep cuda
6 如果出现 -cuda devices 的使用说明则证明配置成功
7 使用方法 在命令行后面添加 -cuda devices
-cuda 0 使用0号GPU运算
-cuda 0,1 创建两个线程 分别运行在0和1号GPU上
-cuda 0,0,1,1 发起四个线程,两个运行在0上,两个运行在1上