是否有一个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
目前没有回答
相关问题 更多 >
编程相关推荐