直接从nvidia GPU内存绘制图形/设置动画

2024-06-06 14:54:48 发布

您现在位置:Python中文网/ 问答频道 /正文

是否有一个python模块用于直接使用GPU内存中的数据绘制图形和/或设置动画?我现在做的是

import numpy as np
import cupy as cp
import matplotlib.pylab as plt

# Create and manipulate data in device RAM
xs = cp.linspace(-3, 3, 500)
ys = cp.exp(xs)

# Copy / move to RAM
xs = cp.asnumpy(xs)
ys = cp.asnumpy(ys)

# Plot and view
plt.plot(xs, ys)
plt.show()

是否可以在不将数据复制到RAM的情况下完成?也许用其他模块? 现在我正在windows 10上使用Anaconda python 3.8.5。我正在使用支持CUDA 11.2的NVIDIA游戏就绪驱动程序(461.09)。我的nvcc编译器和cupy模块的CUDA版本为11.0


Tags: 模块and数据内存importgpuasplt