gu-ma

## pytorch-glumpy.md

      
              1 file
            
          
              0 forks
            
          
              0 comments
            
          
              0 stars
            
          
                gu-ma
                / pytorch-glumpy.md
            
            
              Created
              January 21, 2021 16:11
                — forked from victor-shepardson/pytorch-glumpy.md
            
              
                using pycuda and glumpy to draw pytorch GPU tensors to the screen without copying to host memory
              
          
    Code

https://gist.github.com/victor-shepardson/5b3d3087dc2b4817b9bffdb8e87a57c4
Setup

I'm using Ubuntu 16.04 with a GTX 1060
install CUDA


## pytorch-glumpy.py
from contextlib import contextmanager
import numpy as np
import torch
from torch import Tensor, ByteTensor
import torch.nn.functional as F
from torch.autograd import Variable
import pycuda.driver
from pycuda.gl import graphics_map_flags
from glumpy import app, gloo, gl
	from contextlib import contextmanager
	import numpy as np
	import torch
	from torch import Tensor, ByteTensor
	import torch.nn.functional as F
	from torch.autograd import Variable
	import pycuda.driver
	from pycuda.gl import graphics_map_flags
	from glumpy import app, gloo, gl