使用GPU加速DSPSR

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上

 

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值