分类：cuda处理数据

2021
07-30

pytorch 如何用cuda处理数据

1设置GPU的一些操作设置在os端哪些GPU可见，如果不可见，那肯定是不能够调用的~importosGPU='0,1,2'os.environ['CUDA_VISIBLE_DEVICES']=GPUtorch.cuda.is_available()查看cuda是否可用。iftorch.cuda.is_available():torch.backends.cudnn.benchmark=True'''如果网络的输入数据维度或类型上变化不大，设置torch.backends.cudnn.benchmark=true可以增加运行效率；如果网络的输入数... 继续阅读 >