环境配置
在linux OpenSUSE 42.2系统下运行,安装CUDA Toolkit 10.0,pip中安装pycuda、opencv和numpy。
CUDA
直接从CUDA官网(https://developer.nvidia.com/cuda-90-download-archive)下载对应版本的.run文件,运行后根据提示进行安装。除了“显卡驱动”外所有选项都安装。
安装成功后,应可以通过命令行指令,查看cuda版本。
nvcc --version
pycuda
从pycuda官网(https://mathema.tician.de/software/pycuda/)给出的方法下载,按照帮助文档(https://wiki.tiker.net/PyCuda/Installation/Linux)逐步安装。
安装完成后,可以运行pycuda提供的简单gpu运算代码。
numpy和cv2
正常通过pip安装就可以。
sudo python -m pip install numpy
sudo python -m pip install opencv-python
代码
主要写了