分类：GPU加速

2022
10-31

Python实现GPU加速的基本操作

目录CUDA的线程与块用GPU打印线程编号用GPU打印块编号用GPU打印块的维度用GPU打印线程的维度总结GPU所支持的最大并行度GPU的加速效果总结概要CUDA的线程与块GPU从计算逻辑来讲，可以认为是一个高并行度的计算阵列，我们可以想象成一个二维的像围棋棋盘一样的网格，每一个格子都可以执行一个单独的任务，并且所有的格子可以同时执行计算任务，这就是GPU加速的来源。那么刚才所提到的棋盘，每一列都认为是一个线程，并有自己的线程编... 继续阅读 >

2020
10-08

PyTorch-GPU加速实例

硬件：NVIDIA-GTX1080软件：Windows7、python3.6.5、pytorch-gpu-0.4.1一、基础知识将数据和网络都推到GPU，接上.cuda()二、代码展示importtorchimporttorch.nnasnnimporttorch.utils.dataasDataimporttorchvision#torch.manual_seed(1)EPOCH=1BATCH_SIZE=50LR=0.001DOWNLOAD_MNIST=Falsetrain_data=torchvision.datasets.MNIST(root='./mnist/',train=True,transform=torchvision.transforms.ToTensor(),... 继续阅读 >