什么是 CUDA?
CUDA 是 Nvidia 为并行计算平台和应用程序编程接口创建的一个模型。CUDA 是 NVIDIA 的并行计算架构,通过利用 GPU 的能力,可以大幅度提高计算性能。
CUDA 代码不会在 AMD CPU 或者 Intel HD Graphics 上运行,除非你的机器里有 NVIDIA 的硬件。在 Colab 上,你可以利用 NVIDIA 的 GPU,以及一个预装了 Tensorflow 和其他 ML/DL 工具的功能齐全的 Jupyter note。
第1: 打开colab
然后在runtime-> change type里面修改硬件加速为GPU。
https://colab.research.google.com
第二:检测nvcc版本
一般都是安装了cuda的,直接执行nvcc来查看版本。
3. 安装插件
为了nvcc能运行,需要安装一个插件,识别为cuda代码
!pip install git+https://github.com/andreinechaev/nvcc4jupyter.git
%load_ext nvcc_plugin
4. 运行cuda
%%cu
#include <iostream>
__global__ void cuda_hello(){
printf("Hello World from GPU!\n");
}
int main() {
cuda_hello<<<1,1>>>();
cudaDeviceSynchronize();
return 0;
}
Read more
https://gist.github.com/chasays/81994504fe154f6cd9f799323e331ca8
微信公众号:cdtfug, 欢迎关注一起吹牛逼,也可以加微信号「xiaorik」朋友圈围观。